Skip to content

Commit

Permalink
Ran black
Browse files Browse the repository at this point in the history
  • Loading branch information
Rodot- committed Jul 29, 2024
1 parent 4a82dca commit c441e43
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions tardis/transport/montecarlo/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -103,19 +103,20 @@ def initialize_transport_state(
packet_collection = self.packet_source.create_packets(
no_of_packets, seed_offset=iteration
)


geometry_state = simulation_state.geometry.to_numba()
opacity_state = opacity_state_initialize(
plasma,
self.line_interaction_type,
self.montecarlo_configuration.DISABLE_LINE_SCATTERING,
)
opacity_state = opacity_state[simulation_state.geometry.v_inner_boundary_index: simulation_state.geometry.v_outer_boundary_index]
opacity_state = opacity_state[
simulation_state.geometry.v_inner_boundary_index : simulation_state.geometry.v_outer_boundary_index
]
estimators = initialize_estimator_statistics(
opacity_state.tau_sobolev.shape, gamma_shape
)

transport_state = MonteCarloTransportState(
packet_collection,
estimators,
Expand Down

0 comments on commit c441e43

Please sign in to comment.