code-style bugs fixing

simonsobs · Sep 5, 2024 · 87ad85e · 87ad85e
1 parent aec08a6
commit 87ad85e
Show file tree

Hide file tree

Showing 4 changed files with 163 additions and 93 deletions.
diff --git a/soliket/halo_model/halo_model_fe/cosmology.py b/soliket/halo_model/halo_model_fe/cosmology.py
@@ -26,8 +26,8 @@ def compute_params(self):
 
         D_a = cosmo.angular_diameter_distance(self.redshift).value
         D_h = c / H_0
-        dV_dz = D_h * ((1 + self.redshift) ** 2 * D_a ** 2) / np.sqrt(om_l + 
-                                                                      (1 + self.redshift) ** 3 * om_m) / h ** -3
+        dV_dz = (D_h * ((1 + self.redshift) ** 2 * D_a ** 2) / 
+                 np.sqrt(om_l + (1 + self.redshift) ** 3 * om_m) / h ** -3)
 
         self.om_b = om_b
         self.om_m = om_m

diff --git a/soliket/halo_model/halo_model_fe/lin_matterPS.py b/soliket/halo_model/halo_model_fe/lin_matterPS.py
@@ -22,7 +22,8 @@ def lin_matter_PS(self):
         pars.set_cosmology(H0=par_h, ombh2=par_omega_b, omch2=par_omega_m - par_omega_b, 
                            tau=self.cosmological_param['tau'])
         pars.InitPower.set_params(ns=self.cosmological_param['ns'], 
-                                  As=self.cosmological_param['As'], pivot_scalar=self.cosmological_param['pivot_scalar'])
+                                  As=self.cosmological_param['As'], 
+                                  pivot_scalar=self.cosmological_param['pivot_scalar'])
         pars.set_matter_power(redshifts=self.redshift, kmax=self.kmax)
 
         pars.NonLinear = model.NonLinear_none