add warning for nan or inf alphas

tardis-sn · Nov 1, 2023 · a1f40c8 · a1f40c8
1 parent 46972eb
commit a1f40c8
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/stardis/plasma/base.py b/stardis/plasma/base.py
@@ -232,11 +232,10 @@ def calculate(
             ALPHA_COEFFICIENT * n_lower * linelist.f_lu.values * emission_correction.T
         )
 
-        ###TODO - include warning for missing values.
-        # if np.any(np.isnan(alpha)) or np.any(np.isinf(np.abs(alpha))):
-        #     raise ValueError(
-        #         "Some alpha_line are nan, inf, -inf " " Something went wrong!"
-        #     )
+        if np.any(np.isnan(alpha)) or np.any(np.isinf(np.abs(alpha))):
+            raise ValueError(
+                "Some alpha_line from vald are nan, inf, -inf " " Something went wrong!"
+            )
 
         df = pd.DataFrame(
             alpha.T,