From a68ebaa0c6deca6b4e8908f75f2645c85da50bea Mon Sep 17 00:00:00 2001 From: Andrew Fullard Date: Thu, 15 Aug 2024 15:05:14 -0400 Subject: [PATCH] Black format --- tardis/simulation/base.py | 8 ++++++-- tardis/workflows/simple_simulation.py | 14 +++++++++----- tardis/workflows/standard_simulation.py | 8 ++++++-- 3 files changed, 21 insertions(+), 9 deletions(-) diff --git a/tardis/simulation/base.py b/tardis/simulation/base.py index f7e9a325c22..92a658b80d7 100644 --- a/tardis/simulation/base.py +++ b/tardis/simulation/base.py @@ -291,8 +291,12 @@ def advance_state(self, emitted_luminosity): ) ) - estimated_t_rad = estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature - estimated_dilution_factor = estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + estimated_t_rad = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.temperature + ) + estimated_dilution_factor = ( + estimated_radfield_properties.dilute_blackbody_radiationfield_state.dilution_factor + ) estimated_t_inner = self.estimate_t_inner( self.simulation_state.t_inner, diff --git a/tardis/workflows/simple_simulation.py b/tardis/workflows/simple_simulation.py index 608fb0fbd38..1d4bff8a657 100644 --- a/tardis/workflows/simple_simulation.py +++ b/tardis/workflows/simple_simulation.py @@ -185,8 +185,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, @@ -402,9 +406,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_simulation.py b/tardis/workflows/standard_simulation.py index 429752b543c..ec6c53b1de2 100644 --- a/tardis/workflows/standard_simulation.py +++ b/tardis/workflows/standard_simulation.py @@ -104,8 +104,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,