From 9e5726d609d000661fa9595c4682565b263fcd1c Mon Sep 17 00:00:00 2001
From: Stephane Chavin <stephane.chavin@lis-lab.fr>
Date: Thu, 30 Jan 2025 17:11:56 +0100
Subject: [PATCH] change

---
 get_train_annot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/get_train_annot.py b/get_train_annot.py
index 92ee2b4..0ebe287 100755
--- a/get_train_annot.py
+++ b/get_train_annot.py
@@ -34,9 +34,9 @@ def main(entry, arguments, species_list):
     # Create the list of all possible offset to compute spectrogram
     offset_list = np.arange(
         0, file_duration, arguments.duration - arguments.overlap)
-    new_pos = utils.split_annotations(
+    grp = utils.split_annotations(
         grp, arguments.duration)
-    grp = pd.merge(grp, new_pos)
+    grp = pd.reset_index(drop=True)
     while len(grp) != 0:
         # collect all the data between the offset and duration-overlap
         if len(offset_list) >= 1:
-- 
GitLab