TqdmCallback: fix typo/off-by-one error (#277)

`self.tqdm` should be set if progress_bar is passed.
iterative · Feb 5, 2024 · a7b0eed · a7b0eed
1 parent 542f604
commit a7b0eed
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/dvc_objects/fs/callbacks.py b/src/dvc_objects/fs/callbacks.py
@@ -41,11 +41,10 @@ def __init__(
         from dvc_objects._tqdm import Tqdm
 
         tqdm_kwargs.pop("total", None)
-        tqdm_cls = tqdm_cls or Tqdm
         super().__init__(
-            tqdm_kwargs=tqdm_kwargs, tqdm_cls=tqdm_cls, size=size, value=value
+            tqdm_kwargs=tqdm_kwargs, tqdm_cls=tqdm_cls or Tqdm, size=size, value=value
         )
-        if progress_bar is None:
+        if progress_bar is not None:
             self.tqdm = progress_bar
 
     def branched(self, path_1: "Union[str, BinaryIO]", path_2: str, **kwargs):