|
5 | 5 | import pathlib |
6 | 6 | from typing import Dict, Literal, Optional, Sequence, Tuple, Union |
7 | 7 |
|
8 | | -from rich.progress import ( |
9 | | - BarColumn, |
10 | | - Progress, |
11 | | - SpinnerColumn, |
12 | | - TaskProgressColumn, |
13 | | - TextColumn, |
14 | | -) |
| 8 | +from rich import progress |
15 | 9 |
|
16 | 10 | from wristpy.core import config, exceptions, models |
17 | 11 | from wristpy.io.readers import readers |
@@ -229,14 +223,14 @@ def _run_directory( |
229 | 223 | f"Directory {input} contains no .gt3x or .bin files." |
230 | 224 | ) |
231 | 225 | 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(), |
237 | 231 | console=None, |
238 | | - ) as progress: |
239 | | - task = progress.add_task( |
| 232 | + ) as progress_bar: |
| 233 | + task = progress_bar.add_task( |
240 | 234 | f"[cyan]Processing files in {input.name}...", total=len(file_names) |
241 | 235 | ) |
242 | 236 |
|
@@ -265,7 +259,7 @@ def _run_directory( |
265 | 259 | ) |
266 | 260 | except Exception as e: |
267 | 261 | logger.error("Did not run file: %s, Error: %s", file, e) |
268 | | - progress.update(task, advance=1) |
| 262 | + progress_bar.update(task, advance=1) |
269 | 263 | logger.info("Processing for directory %s completed successfully.", input) |
270 | 264 | return results_dict |
271 | 265 |
|
|
0 commit comments