fix tests (again)

jvshields · Jan 23, 2024 · 857bb92 · 857bb92
1 parent 5cd5151
commit 857bb92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stardis/radiation_field/opacities/opacities_solvers/base.py b/stardis/radiation_field/opacities/opacities_solvers/base.py
@@ -377,10 +377,10 @@ def calc_alpha_line_at_nu(
     if use_vald:
         lines = stellar_plasma.lines_from_linelist
     else:
-        lines = stellar_plasma.lines
+        lines = stellar_plasma.lines.reset_index()
 
         # add ionization energy to lines
-        ionization_data = stellar_plasma.ionization_data
+        ionization_data = stellar_plasma.ionization_data.reset_index()
         ionization_data["ion_number"] -= 1
         lines = pd.merge(
             lines, ionization_data, how="left", on=["atomic_number", "ion_number"]