From 344e015cf20dab1345b72a3ff5e8fc1c84d358a3 Mon Sep 17 00:00:00 2001 From: rodot- Date: Tue, 3 May 2022 11:39:04 -0400 Subject: [PATCH] Second attempt at fixing this, pretty hacky again getting the model attributtes in the plasma but at least it runs --- tardis/model/base.py | 3 +-- tardis/plasma/standard_plasmas.py | 5 ++++- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/tardis/model/base.py b/tardis/model/base.py index e022457daa8..5a64a4d8bcb 100644 --- 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 index b6534b17b3d..45dc82557c2 100644 --- 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