diff --git a/great_ai/src/great_ai/utilities/parallel_map.py b/great_ai/src/great_ai/utilities/parallel_map.py index 108bdcc..38a341e 100644 --- a/great_ai/src/great_ai/utilities/parallel_map.py +++ b/great_ai/src/great_ai/utilities/parallel_map.py @@ -3,7 +3,7 @@ from typing import Any, Callable, Iterable, List, Optional import multiprocess as mp import psutil -from tqdm.auto import tqdm +from tqdm.cli import tqdm from .logger import get_logger