Merge pull request #55 from FRBs/snrdebug

Added check to ensure that FRB SNRs are always above threshold
FRBs · Sep 27, 2024 · b97032f · b97032f
2 parents 6cb23b2 + ed474a7
commit b97032f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/zdm/survey.py b/zdm/survey.py
@@ -957,13 +957,18 @@ def process_survey_file(self,filename:str,
         self.Ss=self.SNRs/self.SNRTHRESHs
         self.TOBS=self.meta['TOBS']
         self.NORM_FRB=self.meta['NORM_FRB']
-        self.Ss[np.where(self.Ss < 1.)[0]]=1
 
         # sets the 'beam' values to unity by default
         self.beam_b=np.array([1])
         self.beam_o=np.array([1])
         self.NBEAMS=1
 
+        # checks for incorrectSNR values
+        toolow = np.where(self.Ss < 1.)[0]
+        if len(toolow) > 0:
+            print("FRBs ",toolow," have SNR < SNRTHRESH!!! Please correct this. Exiting...")
+            exit()
+
         print("FRB survey sucessfully initialised with ",self.NFRB," FRBs starting from", self.iFRB)
 
     def process_dmg(self):