diff --git a/radiant_mlhub/client.py b/radiant_mlhub/client.py index 1ef40e2..80223d5 100644 --- a/radiant_mlhub/client.py +++ b/radiant_mlhub/client.py @@ -8,7 +8,12 @@ import urllib.parse from requests.exceptions import HTTPError -from tqdm import tqdm + +try: + from tqdm.auto import tqdm +except ImportError: # pragma: no cover + # Handles this issue: https://github.com/tqdm/tqdm/issues/1082 + from tqdm import tqdm # type: ignore [no-redef] from .session import get_session from .exceptions import EntityDoesNotExist, MLHubException