add sampling in redshift

examples/injection_examples/change_sampled_parameters.py

@@ -19,7 +19,7 @@ outdir = 'outdir'
 np.random.seed(151226)
 
 injection_parameters = dict(mass_1=36., mass_2=29., a_1=0.4, a_2=0.3, tilt_1=0.5, tilt_2=1.0, phi_12=1.7, phi_jl=0.3,
-                            luminosity_distance=3000., iota=0.4, psi=2.659, phase=1.3, geocent_time=1126259642.413,
+                            luminosity_distance=500, iota=0.4, psi=2.659, phase=1.3, geocent_time=1126259642.413,
                             ra=1.375, dec=-1.2108)
 
 waveform_arguments = dict(waveform_approximant='IMRPhenomPv2',
@@ -30,7 +30,7 @@ waveform_generator = tupak.gw.waveform_generator.WaveformGenerator(
     sampling_frequency=sampling_frequency, time_duration=time_duration,
     frequency_domain_source_model=tupak.gw.source.lal_binary_black_hole,
     parameter_conversion=tupak.gw.conversion.convert_to_lal_binary_black_hole_parameters,
-    non_standard_sampling_parameter_keys=['chirp_mass', 'mass_ratio'],
+    non_standard_sampling_parameter_keys=['chirp_mass', 'mass_ratio', 'redshift'],
     parameters=injection_parameters, waveform_arguments=waveform_arguments)
 hf_signal = waveform_generator.frequency_domain_strain()
 
@@ -43,8 +43,10 @@ IFOs = [tupak.gw.detector.get_interferometer_with_fake_noise_and_injection(
 priors = tupak.gw.prior.BBHPriorSet()
 priors.pop('mass_1')
 priors.pop('mass_2')
+priors.pop('luminosity_distance')
 priors['chirp_mass'] = tupak.prior.Uniform(name='chirp_mass', latex_label='$m_c$', minimum=13, maximum=45)
 priors['mass_ratio'] = tupak.prior.Uniform(name='mass_ratio', latex_label='$q$', minimum=0.1, maximum=1)
+priors['redshift'] = tupak.prior.Uniform(name='redshift', latex_label='$z$', minimum=0, maximum=0.5)
 # These parameters will not be sampled
 for key in ['a_1', 'a_2', 'tilt_1', 'tilt_2', 'phi_12', 'phi_jl', 'psi', 'ra', 'dec', 'geocent_time', 'phase']:
     priors[key] = injection_parameters[key]