diff --git a/pesummary/gw/conversions/snr.py b/pesummary/gw/conversions/snr.py index 38242adc42bb8c7c69d7e4d9c7a57575559ae2ed..f2dc98eec8e610cfde56317f8ad7ff59f73ddf0d 100644 --- a/pesummary/gw/conversions/snr.py +++ b/pesummary/gw/conversions/snr.py @@ -477,7 +477,7 @@ def _prec_ratio_plus_cross(theta_jn, phi_jl, f_plus, f_cross, b_bar): def multipole_snr( mass_1, mass_2, spin_1z, spin_2z, psi, iota, ra, dec, time, distance, phase, f_low=20., f_final=1024., psd={}, approx="IMRPhenomXHM", f_ref=None, - return_data_used=False, multi_process=6, df=1. / 256, + return_data_used=False, multi_process=1, df=1. / 256, multipole=[21, 33, 44], psd_default="aLIGOZeroDetHighPower" ): """Calculate the multipole snr as defined in Mills et al. @@ -524,7 +524,7 @@ def multipole_snr( if True, return a dictionary containing information about what data was used. Default False multi_process: int, optional - The number of cpus to use when computing the precessing_snr. Default 6 + The number of cpus to use when computing the precessing_snr. Default 1 df: float, optional Frequency spacing between frequency samples. Default 1./256 multipole: list, optional @@ -598,7 +598,7 @@ def precessing_snr( mass_1, mass_2, beta, psi_J, a_1, a_2, tilt_1, tilt_2, phi_12, theta_jn, ra, dec, time, phi_jl, distance, phase, f_low=20., psd={}, spin_1z=None, spin_2z=None, chi_eff=None, approx="IMRPhenomPv2", f_final=1024., - f_ref=None, return_data_used=False, multi_process=6, duration=None, + f_ref=None, return_data_used=False, multi_process=1, duration=None, df=1. / 256, psd_default="aLIGOZeroDetHighPower", debug=True ): """Calculate the precessing snr as defined in Fairhurst et al. @@ -660,7 +660,7 @@ def precessing_snr( if True, return a dictionary containing information about what data was used. Default False multi_process: int, optional - The number of cpus to use when computing the precessing_snr. Default 6 + The number of cpus to use when computing the precessing_snr. Default 1 duration: float, optional maximum IMR duration to use to estimate delta_f when PSD is not provided. diff --git a/pesummary/gw/conversions/tgr.py b/pesummary/gw/conversions/tgr.py index e29ef6d1aad1a2324597ef6b3f3f231896a77916..c2f008d4c093964d3e184ea9c4349438d6342afa 100644 --- a/pesummary/gw/conversions/tgr.py +++ b/pesummary/gw/conversions/tgr.py @@ -50,7 +50,7 @@ def _imrct_deviation_parameters_integrand_vectorized( v2, P_final_mass_final_spin_i_interp_object, P_final_mass_final_spin_r_interp_object, - multi_process=4, + multi_process=1, wrapper_function_for_multiprocess=None, ): """Compute the integrand of P(delta_final_mass/final_mass_bar, @@ -71,7 +71,7 @@ def _imrct_deviation_parameters_integrand_vectorized( P_final_mass_final_spin_r_interp_object: interpolated P_r(final_mass, final_spin) multi_process: int - Number of parallel processes. Default: 4 + Number of parallel processes. Default: 1 wrapper_function_for_multiprocess: method Wrapper function for the multiprocessing. Default: None @@ -193,7 +193,7 @@ def _imrct_deviation_parameters_integrand_series( v2, P_final_mass_final_spin_i_interp_object, P_final_mass_final_spin_r_interp_object, - multi_process=4, + multi_process=1, **kwargs ): """ @@ -286,7 +286,7 @@ def imrct_deviation_parameters_from_final_mass_final_spin( N_bins=101, final_mass_deviation_lim=1, final_spin_deviation_lim=1, - multi_process=4, + multi_process=1, use_kde=False, kde=gaussian_kde, kde_kwargs=dict(), @@ -319,7 +319,7 @@ def imrct_deviation_parameters_from_final_mass_final_spin( final_mass_deviation_lim] and [-final_spin_deviation_lim, final_spin_deviation_lim]. Default is 101. multi_process: int - Number of parallel processes. Default is 4. + Number of parallel processes. Default is 1. use_kde: bool If `True`, uses kde instead of interpolation. Default is False. kde: method diff --git a/pesummary/tests/conversion_test.py b/pesummary/tests/conversion_test.py index 727bf90dd47fde777cb0ce117f8d2bb672ccb062..96a03000a1f16e2e5974d60391bb385ee065336d 100644 --- a/pesummary/tests/conversion_test.py +++ b/pesummary/tests/conversion_test.py @@ -770,7 +770,7 @@ class TestPrecessingSNR(object): self.tilt_1, self.tilt_2, self.phi_12, self.theta_jn, self.ra, self.dec, self.time, self.phi_jl, self.distance, self.phase, f_low=self.f_low, spin_1z=self.spin_1z, - spin_2z=self.spin_2z, multi_process=2, debug=False, df=1./8 + spin_2z=self.spin_2z, multi_process=1, debug=False, df=1./8 ) print(rho_p) assert len(rho_p) == len(self.mass_1) @@ -804,7 +804,7 @@ class TestMultipoleSNR(TestPrecessingSNR): rho = multipole_snr( self.mass_1, self.mass_2, self.spin_1z, self.spin_2z, self.psi_l, self.theta_jn, self.ra, self.dec, self.time, self.distance, - self.phase, multi_process=2, df=1./8, multipole=[21, 33, 44] + self.phase, multi_process=1, df=1./8, multipole=[21, 33, 44] ) assert rho.shape[0] == 3 np.testing.assert_almost_equal( @@ -835,7 +835,7 @@ class TestMultipoleSNR(TestPrecessingSNR): rho = multipole_snr( self.mass_1, self.mass_2, self.spin_1z, self.spin_2z, self.psi_l, self.theta_jn, self.ra, self.dec, self.time, self.distance, - self.phase, multi_process=2, df=1./8, multipole=[21, 33, 44, 55] + self.phase, multi_process=1, df=1./8, multipole=[21, 33, 44, 55] )