Merge pull request #15 from ichko/tqdm.auto

Better progress bar for notebooks
hristo-vrigazov · Feb 6, 2024 · b46ae89 · b46ae89
2 parents 7f03de0 + 800de8a
commit b46ae89
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmap_ninja/src/mmap_ninja/base.py b/mmap_ninja/src/mmap_ninja/base.py
@@ -203,7 +203,7 @@ def from_generator_base(out_dir, sample_generator, batch_size, batch_ctor, exten
     samples = []
     memmap = None
     if kwargs.pop("verbose", False):
-        from tqdm import tqdm
+        from tqdm.auto import tqdm
 
         sample_generator = tqdm(sample_generator)
     for sample in sample_generator: