From d2d8c3a57a21e34f379fa0c4fa4a73f36f93f14c Mon Sep 17 00:00:00 2001 From: Moritz Huebner <email@moritz-huebner.de> Date: Fri, 1 Jun 2018 14:18:14 +1000 Subject: [PATCH] Changed the logic around injection into zero noise --- tupak/detector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tupak/detector.py b/tupak/detector.py index f4b2eced8..86ecdb3de 100644 --- a/tupak/detector.py +++ b/tupak/detector.py @@ -260,9 +260,9 @@ class Interferometer(object): logging.warning('Trying to inject signal which is None.') else: signal_ifo = self.get_detector_response(waveform_polarizations, parameters) - try: + if np.shape(self.data).__eq__(np.shape(signal_ifo)): self.data += signal_ifo - except TypeError: + else: logging.info('Injecting into zero noise.') self.data = signal_ifo opt_snr = np.sqrt(tupak.utils.optimal_snr_squared(signal=signal_ifo, interferometer=self, -- GitLab