Fixed divide by zero warning in nu to lambda conversion

tardis-sn · Aug 21, 2020 · 2502110 · 2502110
1 parent 33b7648
commit 2502110
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tardis/analysis.py b/tardis/analysis.py
@@ -41,11 +41,11 @@ def __init__(
             mask
         ]
         self.last_line_interaction_out_angstrom = u.Quantity(
-            output_nu, "Hz"
-        ).to(u.Angstrom, equivalencies=u.spectral())[mask]
+            output_nu[mask], "Hz"
+        ).to(u.Angstrom, equivalencies=u.spectral())
         self.last_line_interaction_in_angstrom = u.Quantity(
-            last_interaction_in_nu, "Hz"
-        ).to(u.Angstrom, equivalencies=u.spectral())[mask]
+            last_interaction_in_nu[mask], "Hz"
+        ).to(u.Angstrom, equivalencies=u.spectral())
         self.lines = lines
 
         self._wavelength_start = 0 * u.angstrom