resolved merge errors

ML4GW · Nov 30, 2023 · b11cc87 · b11cc87
2 parents ae2a708 + e398992
commit b11cc87
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/projects/sandbox/datagen/datagen/utils/injection.py b/projects/sandbox/datagen/datagen/utils/injection.py
@@ -150,11 +150,12 @@ def generate_gw_bns(
         # cut off the first sec of the waveform where the wraparound occurs
         padding_length = padding * sample_rate
         signals[i] = polarizations[:, int(padding_length) :]
-        
+
         # every 1000th waveform
         if not i % 1000:
-            # note the logging.debug so that it's only called if verbose=True.
-            logging.debug(f"{i + 1} polarizations generated") 
+            # note the following is  only called if verbose=True
+            logging.debug(f"{i + 1} polarizations generated")
+
     logging.info("Finished generating polarizations")
     return signals
 

diff --git a/projects/sandbox/train/train/augmentor.py b/projects/sandbox/train/train/augmentor.py
@@ -187,6 +187,7 @@ def sample_responses(self, N: int, kernel_size: int, psds: torch.Tensor):
             responses, _ = self.rescaler(responses, psds**0.5, target_snrs)
 
         kernels = sample_kernels(
+            responses,
             kernel_size=kernel_size,
             max_center_offset=self.trigger_offset,
             coincident=True,