Second attempt at fixing this, pretty hacky again getting the model a…

…ttributtes in the plasma but at least it runs
tardis-sn · May 3, 2022 · 344e015 · 344e015
1 parent c028b33
commit 344e015
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/tardis/model/base.py b/tardis/model/base.py
@@ -186,8 +186,7 @@ def __init__(
             density=density,
         )
         self.raw_abundance = self._abundance
-        self.raw_isotope_abundance = isotope_abundance[self.v_boundary_inner_index + 1 : self.v_boundary_outer_index + 1]
-
+        self.raw_isotope_abundance = isotope_abundance
 
         if t_inner is None:
             if luminosity_requested is not None:

diff --git a/tardis/plasma/standard_plasmas.py b/tardis/plasma/standard_plasmas.py
@@ -251,7 +251,10 @@ def assemble_plasma(config, model, atom_data=None):
 
     if not model.raw_isotope_abundance.empty:
         plasma_modules += isotope_properties
-        kwargs.update(isotope_abundance=model.raw_isotope_abundance)
+        isotope_abundance = model.raw_isotope_abundance.loc[
+            :, model.v_boundary_inner_index : model.v_boundary_outer_index - 1
+        ]
+        kwargs.update(isotope_abundance=isotope_abundance)
 
     kwargs["helium_treatment"] = config.plasma.helium_treatment