diff --git a/bilby/gw/likelihood/base.py b/bilby/gw/likelihood/base.py index 6ed82c7ba6e9789a2a6f0b922d0ee0b12b5cecd7..dca470d2d4946b5ae820ecaa6cff3a019ecb3c08 100644 --- a/bilby/gw/likelihood/base.py +++ b/bilby/gw/likelihood/base.py @@ -1089,6 +1089,7 @@ class GravitationalWaveTransient(Likelihood): waveform_generator_class=self.waveform_generator.__class__, waveform_arguments=self.waveform_generator.waveform_arguments, frequency_domain_source_model=self.waveform_generator.frequency_domain_source_model, + time_domain_source_model=self.waveform_generator.time_domain_source_model, parameter_conversion=self.waveform_generator.parameter_conversion, sampling_frequency=self.waveform_generator.sampling_frequency, duration=self.waveform_generator.duration, diff --git a/bilby/gw/result.py b/bilby/gw/result.py index 6dee3d6bd2424d187280e760af4c04570399026c..154053b2589f114b7b904e38b41be8940ccaedfd 100644 --- a/bilby/gw/result.py +++ b/bilby/gw/result.py @@ -105,6 +105,12 @@ class CompactBinaryCoalescenceResult(CoreResult): return self.__get_from_nested_meta_data( 'likelihood', 'frequency_domain_source_model') + @property + def time_domain_source_model(self): + """ The time domain source model (function)""" + return self.__get_from_nested_meta_data( + 'likelihood', 'time_domain_source_model') + @property def parameter_conversion(self): """ The frequency domain source model (function)""" @@ -381,6 +387,7 @@ class CompactBinaryCoalescenceResult(CoreResult): duration=self.duration, sampling_frequency=self.sampling_frequency, start_time=self.start_time, frequency_domain_source_model=self.frequency_domain_source_model, + time_domain_source_model=self.time_domain_source_model, parameter_conversion=self.parameter_conversion, waveform_arguments=self.waveform_arguments) @@ -589,8 +596,8 @@ class CompactBinaryCoalescenceResult(CoreResult): plot_frequencies, np.percentile(fd_waveforms, lower_percentile, axis=0), np.percentile(fd_waveforms, upper_percentile, axis=0), - color=WAVEFORM_COLOR, label=r'{}\% credible interval'.format( - int(upper_percentile - lower_percentile)), + color=WAVEFORM_COLOR, + label=r'{}% credible interval'.format(int(upper_percentile - lower_percentile)), alpha=0.3) axs[1].plot( plot_times, np.mean(td_waveforms, axis=0), diff --git a/test/gw/likelihood_test.py b/test/gw/likelihood_test.py index eeef6c65c420f740739577fef883e6912e928f63..d404804e97c7c64aa0243b29658d93eb4db6db46 100644 --- a/test/gw/likelihood_test.py +++ b/test/gw/likelihood_test.py @@ -222,6 +222,7 @@ class TestGWTransient(unittest.TestCase): waveform_generator_class=self.waveform_generator.__class__, waveform_arguments=self.waveform_generator.waveform_arguments, frequency_domain_source_model=self.waveform_generator.frequency_domain_source_model, + time_domain_source_model=self.waveform_generator.time_domain_source_model, parameter_conversion=self.waveform_generator.parameter_conversion, sampling_frequency=self.waveform_generator.sampling_frequency, duration=self.waveform_generator.duration,