Add check to make sure heterodyned likelihood frequency grid is correct

kazewong · Jan 3, 2024 · 002fec0 · 002fec0
1 parent de17d53
commit 002fec0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/jimgw/single_event/likelihood.py b/src/jimgw/single_event/likelihood.py
@@ -192,6 +192,10 @@ def __init__(
         self.freq_grid_low = self.freq_grid_low[mask_heterodyne_low]
         self.freq_grid_center = self.freq_grid_center[mask_heterodyne_center]
 
+        # Assure frequency grids have same length
+        if len(self.freq_grid_low) > len(self.freq_grid_center):
+            self.freq_grid_low = self.freq_grid_low[: len(self.freq_grid_center)]
+
         h_sky_low = self.waveform(self.freq_grid_low, self.ref_params)
         h_sky_center = self.waveform(self.freq_grid_center, self.ref_params)