progressbars were set up to take an integer number of iterations

radio-astro-tools · Oct 11, 2024 · c42f3ec · c42f3ec
1 parent 6ac3d04
commit c42f3ec
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/spectral_cube/utils.py b/spectral_cube/utils.py
@@ -6,7 +6,10 @@
 
 bigdataurl = "https://spectral-cube.readthedocs.io/en/latest/big_data.html"
 
-from tqdm.auto import tqdm as ProgressBar
+from tqdm.auto import tqdm
+
+def ProgressBar(niter, **kwargs):
+    return tqdm(total=niter, **kwargs)
 
 
 def cached(func):