Merge branch 'store-time-domain-model-in-meta-data' into 'master'

BUGFIX: Store time domain model in meta data

See merge request !1165

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,

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),

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,