clean up

python3_samples/config_files/default.json

@@ -5,7 +5,7 @@
   "inj_bits": [0, 1, 2, 4],
   "waveform_params_file_name": "waveform_params.ini",
   "max_runtime": 60,
-  "n_injection_samples": 32,
+  "n_injection_samples": 4,
   "n_noise_samples": 16,
   "n_processes": 4,
   "output_file_name": "default.hdf"

python3_samples/generate_sample.py

@@ -3,11 +3,6 @@ The "main script" of this repository: Read in a configuration file and
 generate synthetic GW data according to the provided specifications.
 """
 
-# -----------------------------------------------------------------------------
-# IMPORTS
-# -----------------------------------------------------------------------------
-
-from __future__ import print_function
 
 import argparse
 import numpy as np
@@ -16,8 +11,6 @@ import sys
 import time
 
 from itertools import count
-from multiprocessing import Process, Queue
-from tqdm import tqdm
 
 from utils.configfiles import read_ini_config, read_json_config
 from utils.hdffiles import NoiseTimeline
@@ -265,8 +258,10 @@ if __name__ == '__main__':
 
     # Collect and add injection_parameters (ignore noise samples here, because
     # for those, the injection_parameters are always None)
-    other_keys = ['h1_signal', 'h1_output_signal','h1_snr', 'l1_signal','l1_output_signal', 'l1_snr', 'scale_factor']
+    other_keys = ['h1_signal', 'h1_snr', 'l1_signal', 'l1_snr', 'scale_factor']
+    print(injection_parameters['injection_samples'])
     for key in list(variable_arguments + other_keys):
+        #print(key)
         if injection_parameters['injection_samples']:
             value = np.array([_[key] for _ in
                               injection_parameters['injection_samples']])