Vectorize calc_gamma_van_der_waals()

smokestacklightnin · Aug 17, 2023 · 58e4fea · 58e4fea
1 parent d71f027
commit 58e4fea
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/stardis/opacities/broadening.py b/stardis/opacities/broadening.py
@@ -201,7 +201,7 @@ def calc_gamma_quadratic_stark(
 
 
 @numba.njit
-def calc_gamma_van_der_waals(
+def _calc_gamma_van_der_waals(
     ion_number,
     n_eff_upper,
     n_eff_lower,
@@ -260,6 +260,25 @@ def calc_gamma_van_der_waals(
     return gamma_van_der_waals
 
 
+@numba.vectorize(nopython=True)
+def calc_gamma_van_der_waals(
+    ion_number,
+    n_eff_upper,
+    n_eff_lower,
+    temperature,
+    h_density,
+    h_mass,
+):
+    return _calc_gamma_van_der_waals(
+        ion_number,
+        n_eff_upper,
+        n_eff_lower,
+        temperature,
+        h_density,
+        h_mass,
+    )
+
+
 @numba.njit
 def calc_gamma(
     atomic_number,