diff --git a/bilby/gw/detector/networks.py b/bilby/gw/detector/networks.py index d354f2bf48a9f98c2d21b981fe7179affca7d3ac..4370219c183e2a20e86ad53fece558cdd147f3a4 100644 --- a/bilby/gw/detector/networks.py +++ b/bilby/gw/detector/networks.py @@ -109,7 +109,13 @@ class InterferometerList(list): duration=duration, start_time=start_time) - def inject_signal(self, parameters=None, injection_polarizations=None, waveform_generator=None): + def inject_signal( + self, + parameters=None, + injection_polarizations=None, + waveform_generator=None, + raise_error=True, + ): """ Inject a signal into noise in each of the three detectors. Parameters @@ -124,6 +130,9 @@ class InterferometerList(list): waveform_generator: bilby.gw.waveform_generator.WaveformGenerator A WaveformGenerator instance using the source model to inject. If `injection_polarizations` is given, this will be ignored. + raise_error: bool + Whether to raise an error if the injected signal does not fit in + the segment. Notes ========== @@ -148,7 +157,12 @@ class InterferometerList(list): all_injection_polarizations = list() for interferometer in self: all_injection_polarizations.append( - interferometer.inject_signal(parameters=parameters, injection_polarizations=injection_polarizations)) + interferometer.inject_signal( + parameters=parameters, + injection_polarizations=injection_polarizations, + raise_error=raise_error, + ) + ) return all_injection_polarizations diff --git a/test/gw/detector/networks_test.py b/test/gw/detector/networks_test.py index 4484cfb12d526ecb11940e390e78b2e10c9c21c7..2ad4060e5934a571e9b902eb22119a9303a80cf9 100644 --- a/test/gw/detector/networks_test.py +++ b/test/gw/detector/networks_test.py @@ -240,8 +240,8 @@ class TestInterferometerList(unittest.TestCase): @patch.object(bilby.gw.detector.Interferometer, "inject_signal") def test_inject_signal_with_inj_pol(self, m): - self.ifo_list.inject_signal(injection_polarizations=dict(plus=1)) - m.assert_called_with(parameters=None, injection_polarizations=dict(plus=1)) + self.ifo_list.inject_signal(injection_polarizations=dict(plus=1), raise_error=False) + m.assert_called_with(parameters=None, injection_polarizations=dict(plus=1), raise_error=False) self.assertEqual(len(self.ifo_list), m.call_count) @patch.object(bilby.gw.detector.Interferometer, "inject_signal")