fix benchmarks

tardis-sn · Jun 11, 2024 · 74679db · 74679db
1 parent 662c785
commit 74679db
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/benchmarks/run_stardis.py b/benchmarks/run_stardis.py
@@ -57,9 +57,7 @@ def setup(self):
                 1,
                 np.min(
                     [
-                        len(
-                            stellar_model.composition.atomic_mass_fraction.columns.tolist()
-                        ),
+                        len(stellar_model.composition.element_masses),
                         config.model.final_atomic_number,
                     ]
                 )
@@ -168,9 +166,7 @@ def setup(self):
                 1,
                 np.min(
                     [
-                        len(
-                            stellar_model.composition.atomic_mass_fraction.columns.tolist()
-                        ),
+                        len(stellar_model.composition.element_masses),
                         config.model.final_atomic_number,
                     ]
                 )