diff --git a/tardis/workflows/simple_tardis_workflow.py b/tardis/workflows/simple_tardis_workflow.py index b14708247d8..c7d0649eb11 100644 --- a/tardis/workflows/simple_tardis_workflow.py +++ b/tardis/workflows/simple_tardis_workflow.py @@ -161,8 +161,12 @@ def get_convergence_estimates(self, transport_state): ) ) - estimated_t_radiative = estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature - estimated_dilution_factor = estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + estimated_t_radiative = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature + ) + estimated_dilution_factor = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + ) emitted_luminosity = calculate_filtered_luminosity( transport_state.emitted_packet_nu, @@ -380,9 +384,9 @@ def initialize_spectrum_solver( self.spectrum_solver.transport_state = transport_state if virtual_packet_energies is not None: - self.spectrum_solver._montecarlo_virtual_luminosity.value[:] = ( - virtual_packet_energies - ) + self.spectrum_solver._montecarlo_virtual_luminosity.value[ + : + ] = virtual_packet_energies if self.integrated_spectrum_settings is not None: # Set up spectrum solver integrator diff --git a/tardis/workflows/standard_tardis_workflow.py b/tardis/workflows/standard_tardis_workflow.py index f5842943f32..4e4e2d0de32 100644 --- a/tardis/workflows/standard_tardis_workflow.py +++ b/tardis/workflows/standard_tardis_workflow.py @@ -130,8 +130,12 @@ def get_convergence_estimates(self, transport_state): ) ) - estimated_t_radiative = estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature - estimated_dilution_factor = estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + estimated_t_radiative = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature + ) + estimated_dilution_factor = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + ) emitted_luminosity = calculate_filtered_luminosity( transport_state.emitted_packet_nu,