From c94e04ae556375b2a16f7a5d939b5d802d41f74e Mon Sep 17 00:00:00 2001
From: Marek Blok <Marek.Blok@pg.edu.pl>
Date: Sat, 4 Mar 2023 18:14:34 +0100
Subject: [PATCH] fixed get_SNR_dB SNR_dB_range infterpretation for int and
float scallar values
---
data_processor/creators.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/data_processor/creators.py b/data_processor/creators.py
index bded5a1..5b2d568 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]
--
GitLab