Revert .values change

tardis-sn · Sep 19, 2022 · eb8a42c · eb8a42c
1 parent b84cc84
commit eb8a42c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tardis/plasma/properties/nlte.py b/tardis/plasma/properties/nlte.py
@@ -86,7 +86,7 @@ def calculate(
         helium_population.loc[0, 0] = 0.0
         # He II excited states
         he_two_population = level_boltzmann_factor.loc[2, 1].mul(
-            (g.loc[2,1,0].values ** (-1.0))
+            (float(g.loc[2,1,0]) ** (-1.0))
         )
         helium_population.loc[1].update(he_two_population)
         # He II ground state
@@ -117,7 +117,7 @@ def calculate_helium_one(
         """
         return (
             level_boltzmann_factor.loc[2, 0]
-            * (1.0 / (g.loc[2,1,0].values * 2))
+            * (1.0 / (float(g.loc[2,1,0]) * 2))
             * (1 / g_electron)
             * (1 / (w**2.0))
             * np.exp(ionization_data.loc[2, 1] * beta_rad)