Make tqdm work nicely in notebooks

tskit-dev · Feb 22, 2024 · 91c1575 · 91c1575
1 parent e99a3dc
commit 91c1575
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,