progress: remove unused logging/tqdm lock

src/datachain/progress.py

@@ -1,14 +1,5 @@
-"""Manages progress bars."""
-
-import logging
-from threading import RLock
-
 from fsspec import Callback
 from fsspec.callbacks import TqdmCallback
-from tqdm.auto import tqdm
-
-logger = logging.getLogger(__name__)
-tqdm.set_lock(RLock())
 
 
 class CombinedDownloadCallback(Callback):
@@ -24,10 +15,6 @@ def increment_file_count(self, n: int = 1) -> None:
 class TqdmCombinedDownloadCallback(CombinedDownloadCallback, TqdmCallback):
     def __init__(self, tqdm_kwargs=None, *args, **kwargs):
         self.files_count = 0
-        tqdm_kwargs = tqdm_kwargs or {}
-        tqdm_kwargs.setdefault("postfix", {}).setdefault("files", self.files_count)
-        kwargs = kwargs or {}
-        kwargs["tqdm_cls"] = tqdm
         super().__init__(tqdm_kwargs, *args, **kwargs)
 
     def increment_file_count(self, n: int = 1) -> None:

src/datachain/query/dataset.py

@@ -355,14 +355,15 @@ def process_udf_outputs(
 
 def get_download_callback(suffix: str = "", **kwargs) -> CombinedDownloadCallback:
     return TqdmCombinedDownloadCallback(
-        {
+        tqdm_kwargs={
             "desc": "Download" + suffix,
             "unit": "B",
             "unit_scale": True,
             "unit_divisor": 1024,
             "leave": False,
             **kwargs,
         },
+        tqdm_cls=tqdm,
     )