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