Commit 18514b44 authored by Yifan Wang's avatar Yifan Wang
Browse files

clean up

parent ea69739b
......@@ -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"
......
......@@ -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']])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment