From b88b2b6247e3713ff0e73f7e8cd3630965d2f4ef Mon Sep 17 00:00:00 2001
From: sprause <sprause95@gmail.com>
Date: Wed, 25 Jan 2023 10:18:20 +0100
Subject: [PATCH] adjusted check of length of on- and offests after correction
---
code/modules/behaviour_handling.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/code/modules/behaviour_handling.py b/code/modules/behaviour_handling.py
index 6641309..29499b6 100644
--- a/code/modules/behaviour_handling.py
+++ b/code/modules/behaviour_handling.py
@@ -86,14 +86,19 @@ def correct_chasing_events(
category = np.delete(category, wrong_bh)
timestamps = np.delete(timestamps, wrong_bh)
+ new_onset_ids = np.arange(
+ len(category))[category == 0]
+ new_offset_ids = np.arange(
+ len(category))[category == 1]
+
# Check whether on- or offset is longer and calculate length difference
- if len(onset_ids) > len(offset_ids):
+ if len(new_onset_ids) > len(new_offset_ids):
len_diff = len(onset_ids) - len(offset_ids)
logger.info(f'Onsets are greater than offsets by {len_diff}')
- elif len(onset_ids) < len(offset_ids):
+ elif len(new_onset_ids) < len(new_offset_ids):
len_diff = len(offset_ids) - len(onset_ids)
logger.info(f'Offsets are greater than onsets by {len_diff}')
- elif len(onset_ids) == len(offset_ids):
+ elif len(new_onset_ids) == len(new_offset_ids):
logger.info('Chasing events are equal')
return category, timestamps