diff --git a/vision_engine/input/series_paths_loader.py b/vision_engine/input/series_paths_loader.py index c0aedfb..b6ba67c 100644 --- a/vision_engine/input/series_paths_loader.py +++ b/vision_engine/input/series_paths_loader.py @@ -7,6 +7,7 @@ import os from typing import List from pathlib import Path +from tqdm import tqdm from ..core.data_structures import SeriesInfo from ..core.exceptions import InputError @@ -61,7 +62,7 @@ def get_series_list(self, csv_path: str) -> List[SeriesInfo]: series_list = [] path_column = getattr(self.config, "path_column", "series_path") - for idx, row in df.iterrows(): + for idx, row in tqdm(df.iterrows(), total=len(df), desc='Gathering series info'): try: series_path = str(row[path_column])