Warn for KeyError in get_decompressed_waveform

gwastro · Jan 16, 2025 · 1d3ac81 · 1d3ac81
1 parent fb5bbc3
commit 1d3ac81
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/pycbc/waveform/bank.py b/pycbc/waveform/bank.py
@@ -40,6 +40,7 @@
 import pycbc.io
 from pycbc.io.ligolw import LIGOLWContentHandler
 import hashlib
+import warnings
 
 
 def sigma_cached(self, psd):
@@ -782,7 +783,14 @@ def generate_with_delta_f_and_max_freq(self, t_num, max_freq, delta_f,
                                                     f_lower=low_frequency_cutoff,
                                                     approximant=approximant,
                                                     df=delta_f)
-        except (ValueError, KeyError) :
+        except (ValueError, KeyError) as e:
+            if isinstance(e, KeyError):
+                warnings.warn(
+                    "self.get_decompressed_waveform has raised a KeyError. "
+                    "This may be as the compressed waveform has not been "
+                    "generated for this approximant, but it could indicate "
+                    "a more serious issue. Approximant: %s" % approximant
+                )
             htilde = pycbc.waveform.get_waveform_filter(
                 cached_mem, self.table[t_num], approximant=approximant,
                 f_lower=low_frequency_cutoff, f_final=max_freq, delta_f=delta_f,
@@ -821,7 +829,14 @@ def __getitem__(self, index):
             htilde = self.get_decompressed_waveform(tempout, index, f_lower=f_low,
                                                     approximant=approximant, df=None)
 
-        except (ValueError, KeyError) :
+        except (ValueError, KeyError) as e:
+            if isinstance(e, KeyError):
+                warnings.warn(
+                    "self.get_decompressed_waveform has raised a KeyError. "
+                    "This may be as the compressed waveform has not been "
+                    "generated for this approximant, but it could indicate "
+                    "a more serious issue. Approximant: %s" % approximant
+                )
             htilde = pycbc.waveform.get_waveform_filter(
                 tempout[0:self.filter_length], self.table[index],
                 approximant=approximant, f_lower=f_low, f_final=f_end,