ENH: optional tqdm

gallantlab · Jun 17, 2021 · 6fd403f · 6fd403f
1 parent 101e87e
commit 6fd403f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/moten/extras.py b/moten/extras.py
@@ -7,8 +7,12 @@
                          pointwise_square,
                          )
 
-with_tqdm = False
-
+# Make tqdm optional
+with_tqdm = True
+try:
+    from tqdm import tqdm
+except ImportError:
+    with_tqdm = False
 
 def pixbypix_covariance_from_frames_generator(data_generator,
                                               batch_size=1000,