add tqdm status to plot_pfss()

jgieseler · Aug 15, 2023 · 3c1b580 · 3c1b580
1 parent 25bfeb2
commit 3c1b580
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/solarmach/__init__.py b/solarmach/__init__.py
@@ -967,7 +967,7 @@ def plot_pfss(self,
         lon_sep_angles = np.array([])
         lat_sep_angles = np.array([])
 
-        for i, body_id in enumerate(self.body_dict):
+        for i, body_id in enumerate(tqdm(self.body_dict)):
 
             body_lab = self.body_dict[body_id][1]
             body_color = self.body_dict[body_id][2]
@@ -1554,3 +1554,9 @@ def _isstreamlit():
     except ModuleNotFoundError:
         use_streamlit = False
     return use_streamlit
+
+
+if _isstreamlit():
+    from stqdm import stqdm as tqdm
+else:
+    from tqdm.auto import tqdm