Skip to content

Commit

Permalink
Black format
Browse files Browse the repository at this point in the history
  • Loading branch information
andrewfullard committed Aug 15, 2024
1 parent 011c415 commit a68ebaa
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 9 deletions.
8 changes: 6 additions & 2 deletions tardis/simulation/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down
14 changes: 9 additions & 5 deletions tardis/workflows/simple_simulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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
Expand Down
8 changes: 6 additions & 2 deletions tardis/workflows/standard_simulation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit a68ebaa

Please sign in to comment.