From fd94363bd61d3aba9c50a20137106d2aec26e44f Mon Sep 17 00:00:00 2001 From: Josh Shields Date: Thu, 1 Feb 2024 14:07:57 -0500 Subject: [PATCH] remove for alpha_electron for loop --- stardis/radiation_field/opacities/opacities_solvers/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/stardis/radiation_field/opacities/opacities_solvers/base.py b/stardis/radiation_field/opacities/opacities_solvers/base.py index 0bf796e1..37c4bac5 100644 --- a/stardis/radiation_field/opacities/opacities_solvers/base.py +++ b/stardis/radiation_field/opacities/opacities_solvers/base.py @@ -170,9 +170,9 @@ def calc_alpha_electron( const.sigma_T.cgs.value * stellar_plasma.electron_densities.values ) - alpha_electron = np.zeros((stellar_model.no_of_depth_points, len(tracing_nus))) - for j in range(stellar_model.no_of_depth_points): - alpha_electron[j] = alpha_electron_by_depth_point[j] + alpha_electron = np.repeat( + alpha_electron_by_depth_point[:, np.newaxis], len(tracing_nus), axis=1 + ) return alpha_electron