Merge pull request #906 from hyanwong/tqdm-notebook

Make tqdm work nicely in notebooks
tskit-dev · Mar 11, 2024 · 04e9766 · 04e9766
2 parents d9e58cc + 4ffb27d
commit 04e9766
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tsinfer/progress.py b/tsinfer/progress.py
@@ -19,7 +19,7 @@
 """
 A progress monitor class for tsinfer
 """
-import tqdm
+from tqdm.auto import tqdm
 
 
 class ProgressMonitor:
@@ -83,7 +83,7 @@ def get(self, key, total):
             "{desc}{percentage:3.0f}%|{bar}"
             "| {n_fmt}/{total_fmt} [{elapsed}, {rate_fmt}{postfix}]"
         )
-        self.current_instance = tqdm.tqdm(
+        self.current_instance = tqdm(
             desc=desc,
             total=total,
             disable=not self.enabled,