Use the correct radiation_field_state attribute

andrewfullard · Aug 15, 2024 · 8b3ad36 · 8b3ad36
1 parent 41e8f32
commit 8b3ad36
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tardis/simulation/base.py b/tardis/simulation/base.py
@@ -275,9 +275,8 @@ def advance_state(self, emitted_luminosity):
         -------
             converged : bool
         """
-        current_radiation_field_state = DilutePlanckianRadiationField(
-            self.simulation_state.t_radiative,
-            self.simulation_state.dilution_factor,
+        current_radiation_field_state = (
+            self.simulation_state.radiation_field_state
         )
 
         estimated_radfield_properties = (
@@ -351,8 +350,7 @@ def advance_state(self, emitted_luminosity):
             next_dilution_factor,
             next_t_inner,
         )
-        self.simulation_state.t_radiative = next_t_radiative
-        self.simulation_state.dilution_factor = next_dilution_factor
+        self.simulation_state.radiation_field_state = next_radiation_field_state
         self.simulation_state.blackbody_packet_source.temperature = next_t_inner
 
         update_properties = dict(