Normalizing dataframe after merging

tardis-sn · Oct 31, 2022 · 3268459 · 3268459
1 parent 10674b4
commit 3268459
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tardis/model/base.py b/tardis/model/base.py
@@ -262,6 +262,8 @@ def __init__(
         isotope_mass_fraction = self.raw_isotope_abundance.decay(13).add(
             raw_abundance, fill_value=0
         )
+        norm_factor = isotope_mass_fraction.sum(axis=0)
+        isotope_mass_fraction /= norm_factor
 
         atomic_mass = None
         if elemental_mass is not None: