Enable tqdm progress in cloud environments

src/huggingface_hub/file_download.py

@@ -62,6 +62,7 @@
 from .utils._runtime import _PY_VERSION  # noqa: F401 # for backward compatibility
 from .utils._typing import HTTP_METHOD_T
 from .utils.sha import sha_fileobj
+from .utils.tqdm import is_tqdm_disabled
 
 
 logger = logging.get_logger(__name__)
@@ -402,9 +403,7 @@ def http_get(
             total=total,
             initial=resume_size,
             desc=displayed_filename,
-            disable=True if (logger.getEffectiveLevel() == logging.NOTSET) else None,
-            # ^ set `disable=None` rather than `disable=False` by default to disable progress bar when no TTY attached
-            # see https://github.com/huggingface/huggingface_hub/pull/2000
+            disable=is_tqdm_disabled(logger.getEffectiveLevel()),
             name="huggingface_hub.http_get",
         )
         if _tqdm_bar is None

src/huggingface_hub/lfs.py

@@ -39,6 +39,7 @@
 )
 from .utils._lfs import SliceFileObj
 from .utils.sha import sha256, sha_fileobj
+from .utils.tqdm import is_tqdm_disabled
 
 
 if TYPE_CHECKING:
@@ -430,17 +431,13 @@ def _upload_parts_hf_transfer(
     if len(desc) > 40:
         desc = f"(…){desc[-40:]}"
 
-    # set `disable=None` rather than `disable=False` by default to disable progress bar when no TTY attached
-    # see https://github.com/huggingface/huggingface_hub/pull/2000
-    disable = True if (logger.getEffectiveLevel() == logging.NOTSET) else None
-
     with tqdm(
         unit="B",
         unit_scale=True,
         total=total,
         initial=0,
         desc=desc,
-        disable=disable,
+        disable=is_tqdm_disabled(logger.getEffectiveLevel()),
         name="huggingface_hub.lfs_upload",
     ) as progress:
         try:

src/huggingface_hub/utils/tqdm.py

@@ -81,6 +81,8 @@
 """
 
 import io
+import logging
+import os
 import warnings
 from contextlib import contextmanager
 from pathlib import Path
@@ -196,6 +198,19 @@ def are_progress_bars_disabled(name: Optional[str] = None) -> bool:
     return not progress_bar_states.get("_global", True)
 
 
+def is_tqdm_disabled(log_level: int) -> Optional[bool]:
+    """
+    Determine if tqdm progress bars should be disabled based on logging level and environment settings.
+
+    see https://github.com/huggingface/huggingface_hub/pull/2000 and https://github.com/huggingface/huggingface_hub/pull/2698.
+    """
+    if log_level == logging.NOTSET:
+        return True
+    if os.getenv("TQDM_POSITION") == "-1":
+        return False
+    return None
+
+
 class tqdm(old_tqdm):
     """
     Class to override `disable` argument in case progress bars are globally disabled.