Abundance widget bug fix (#2577)

* Fixed CustomAbundanceWidget Bug caused by using wrong parameters of simulation_state * Reformatted using black suggestion * Updated Documentation concerning usage of CustomAbundanceWidget.from_simulation
tardis-sn · Apr 24, 2024 · 63cde91 · 63cde91
1 parent 760dacd
commit 63cde91
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/docs/io/visualization/how_to_abundance_widget.ipynb b/docs/io/visualization/how_to_abundance_widget.ipynb
@@ -97,7 +97,7 @@
    "outputs": [],
    "source": [
     "# sim = run_tardis(\"tardis_example.yml\")\n",
-    "# widget = CustomAbundanceWidget.from_sim(sim)"
+    "# widget = CustomAbundanceWidget.from_simulation(sim)"
    ]
   },
   {

diff --git a/tardis/visualization/widgets/custom_abundance.py b/tardis/visualization/widgets/custom_abundance.py
@@ -281,8 +281,10 @@ def from_simulation(cls, sim):
         -------
         CustomAbundanceWidgetData
         """
-        abundance = sim.simulation_state.raw_abundance.copy()
-        isotope_abundance = sim.simulation_state.raw_isotope_abundance.copy()
+        abundance = sim.simulation_state.abundance.copy()
+        isotope_abundance = (
+            sim.simulation_state.composition.raw_isotope_abundance.copy()
+        )
 
         # integrate element and isotope to one DataFrame
         abundance["mass_number"] = ""