From bac3de6af8a8a3d1fe52947120c51f9381f63c32 Mon Sep 17 00:00:00 2001
From: pswain <peter.swain@ed.ac.uk>
Date: Tue, 28 Nov 2023 15:58:33 +0000
Subject: [PATCH] bug(apply_merges): merges implementing on more than
valid_merges
---
src/agora/utils/merge.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/agora/utils/merge.py b/src/agora/utils/merge.py
index 183acbde..dd2dfeec 100644
--- a/src/agora/utils/merge.py
+++ b/src/agora/utils/merge.py
@@ -130,8 +130,8 @@ def apply_merges(data: pd.DataFrame, merges: np.ndarray):
# merge tracks
if valid_merges.any():
to_merge = data.loc[valid_indices].copy()
- left_indices = merges[:, 0]
- right_indices = merges[:, 1]
+ left_indices = merges[valid_merges, 0]
+ right_indices = merges[valid_merges, 1]
# join left track with right track
for left_index, right_index in zip(left_indices, right_indices):
to_merge.loc[tuple(left_index)] = join_two_tracks(
--
GitLab