From 29f693554ed58a17da3786d6c8b0572c81c9f6aa Mon Sep 17 00:00:00 2001 From: Till Raab Date: Fri, 20 Oct 2023 12:58:44 +0200 Subject: [PATCH] bf --- data/generate_dataset.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/data/generate_dataset.py b/data/generate_dataset.py index caa01cf..b5d58a5 100644 --- a/data/generate_dataset.py +++ b/data/generate_dataset.py @@ -143,13 +143,11 @@ def main(args): left_time_bound = times_v[rise_idx_oi] right_time_bound = np.zeros_like(left_time_bound) - for enu, Ct_idx in enumerate(times_v[rise_idx_oi]): + for enu, Ct_oi in enumerate(times_v[rise_idx_oi]): Crise_size = rise_size_oi[enu] Cblf = closest_baseline_freq[enu] - embed() - quit() - rise_end_t = times_v[(times_v[idx_v] > Ct_idx) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)] + rise_end_t = times_v[(times_v > Ct_oi) & (fish_freq[id_idx] < Cblf + Crise_size * 0.37)] if len(rise_end_t) == 0: right_time_bound[enu] = np.nan else: