replace extra condition with mask

tardis-sn · Oct 10, 2024 · 20f46d1 · 20f46d1
1 parent 8a145a3
commit 20f46d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tardis/plasma/properties/atomic.py b/tardis/plasma/properties/atomic.py
@@ -788,9 +788,9 @@ def exp1_times_exp(x):
             Output array.
         """
         x = np.asarray(x)
-        f = exp1(x) * np.exp(x, where=x <= 500)
-        # Use Laurent series for large values to avoid infinite exponential
         mask = x > 500
+        f = exp1(x) * np.exp(x * (~mask))
+        # Use Laurent series for large values to avoid infinite exponential
         f[mask] = (x**-1 - x**-2 + 2 * x**-3 - 6 * x**-4)[mask]
         return f