diff --git a/.github/workflows/codestyle.yml b/.github/workflows/codestyle.yml index a2660c172e3..9ec58ea12d8 100644 --- a/.github/workflows/codestyle.yml +++ b/.github/workflows/codestyle.yml @@ -22,6 +22,16 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v4 + if: github.event_name == 'push' + with: + fetch-depth: 0 + + - name: Checkout PR and master branch + if: github.event_name == 'pull_request_target' + uses: actions/checkout@v4 + with: + ref: ${{ github.event.pull_request.head.sha }} + fetch-depth: 0 - name: Setup Python uses: actions/setup-python@v5 diff --git a/tardis/opacities/macro_atom/macroatom_solver.py b/tardis/opacities/macro_atom/macroatom_solver.py index 9e306d3dff6..8f92602abf6 100644 --- a/tardis/opacities/macro_atom/macroatom_solver.py +++ b/tardis/opacities/macro_atom/macroatom_solver.py @@ -33,9 +33,7 @@ def initialize_transition_probabilities(self, atomic_data): Atomic Data """ - coef_and_block_ref = initialize_transition_probabilities( - atomic_data - ) + coef_and_block_ref = initialize_transition_probabilities(atomic_data) self.transition_probability_coef = coef_and_block_ref[ "transition_probability_coef" ] @@ -69,17 +67,15 @@ def solve_transition_probabilities( if self.initialize: self.initialize_transition_probabilities(atomic_data) - transition_probabilities = ( - calculate_transition_probabilities( - atomic_data, - legacy_plasma.beta_sobolev, - legacy_plasma.j_blues, - stimulated_emission_factor, - tau_sobolev, - self.transition_probability_coef, - self.block_references, - normalize=self.normalize, - ) + transition_probabilities = calculate_transition_probabilities( + atomic_data, + legacy_plasma.beta_sobolev, + legacy_plasma.j_blues, + stimulated_emission_factor, + tau_sobolev, + self.transition_probability_coef, + self.block_references, + normalize=self.normalize, ) return transition_probabilities @@ -109,13 +105,11 @@ def solve( State of the macro atom ready to be placed into the OpacityState """ - transition_probabilities = ( - self.solve_transition_probabilities( - atomic_data, - legacy_plasma, - tau_sobolev, - stimulated_emission_factor, - ) + transition_probabilities = self.solve_transition_probabilities( + atomic_data, + legacy_plasma, + tau_sobolev, + stimulated_emission_factor, ) macro_block_references = atomic_data.macro_atom_references[ diff --git a/tardis/opacities/opacity_solver.py b/tardis/opacities/opacity_solver.py index 3fee278b613..4561984bcd8 100644 --- a/tardis/opacities/opacity_solver.py +++ b/tardis/opacities/opacity_solver.py @@ -6,8 +6,6 @@ import pandas as pd - - class OpacitySolver(object): line_interaction_type: str = "scatter"