Skip to content

Commit 76ead3e

Browse files
committed
Edits for PR
1 parent 6a2dc80 commit 76ead3e

3 files changed

Lines changed: 14 additions & 19 deletions

File tree

poetry.lock

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ pydantic = "^2.11.1"
2424
pydantic-settings = "^2.3.4"
2525
scipy = "^1.14.1"
2626
typer = "^0.15.2"
27+
rich = "^14.1.0"
2728

2829
[tool.poetry.group.dev.dependencies]
2930
pytest = "^8.0.0"

src/wristpy/core/orchestrator.py

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,7 @@
55
import pathlib
66
from typing import Dict, Literal, Optional, Sequence, Tuple, Union
77

8-
from rich.progress import (
9-
BarColumn,
10-
Progress,
11-
SpinnerColumn,
12-
TaskProgressColumn,
13-
TextColumn,
14-
)
8+
from rich import progress
159

1610
from wristpy.core import config, exceptions, models
1711
from wristpy.io.readers import readers
@@ -229,14 +223,14 @@ def _run_directory(
229223
f"Directory {input} contains no .gt3x or .bin files."
230224
)
231225
results_dict = {}
232-
with Progress(
233-
SpinnerColumn(),
234-
TextColumn("[progress.description]{task.description}"),
235-
BarColumn(),
236-
TaskProgressColumn(),
226+
with progress.Progress(
227+
progress.SpinnerColumn(),
228+
progress.TextColumn("[progress.description]{task.description}"),
229+
progress.BarColumn(),
230+
progress.TaskProgressColumn(),
237231
console=None,
238-
) as progress:
239-
task = progress.add_task(
232+
) as progress_bar:
233+
task = progress_bar.add_task(
240234
f"[cyan]Processing files in {input.name}...", total=len(file_names)
241235
)
242236

@@ -265,7 +259,7 @@ def _run_directory(
265259
)
266260
except Exception as e:
267261
logger.error("Did not run file: %s, Error: %s", file, e)
268-
progress.update(task, advance=1)
262+
progress_bar.update(task, advance=1)
269263
logger.info("Processing for directory %s completed successfully.", input)
270264
return results_dict
271265

0 commit comments

Comments
 (0)