diff --git a/data_processor/creators.py b/data_processor/creators.py index bded5a12e1e2661819c3f33d856a1553061e6f81..5b2d568177ae7ad7248512758b1b5edfc0c5e04c 100644 --- a/data_processor/creators.py +++ b/data_processor/creators.py @@ -88,8 +88,9 @@ def get_SNR_dB(rng_state_filename, SNR_index, SNR_dB_range, vs = None): SNR_rng_state.init(vs.rng_seed, SNR_index) # randomize SNR from given range - if type(SNR_dB_range) == np.float64: - SNR_dB = SNR_dB_range + #if (type(SNR_dB_range) == np.float64) or (type(SNR_dB_range) == float): + if not hasattr(SNR_dB_range, "__len__"): + SNR_dB = np.float64(SNR_dB_range) else: if SNR_dB_range[0] == SNR_dB_range[1]: SNR_dB = SNR_dB_range[0]