progressbars were set up to take an integer number of iterations

radio-astro-tools · Oct 13, 2024 · 0fd643e · 0fd643e
1 parent 565986e
commit 0fd643e
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):