From 000d7354d794d7230f3b4ebd9cc56f7837a67e6f Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Wed, 9 Dec 2020 13:21:21 +0100
Subject: [PATCH] Simplify code
---
km3io/tools.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/km3io/tools.py b/km3io/tools.py
index 27fa1d4..2bc0e00 100644
--- a/km3io/tools.py
+++ b/km3io/tools.py
@@ -276,7 +276,7 @@ def best_track(tracks, startend=None, minmax=None, stages=None):
tracks = tracks[m1]
- rec_stage_lengths = ak.num(tracks.rec_stages, axis=axis + 1)
+ rec_stage_lengths = ak.num(tracks.rec_stages, axis=-1)
max_rec_stage_length = ak.max(rec_stage_lengths, axis=axis)
m2 = rec_stage_lengths == max_rec_stage_length
tracks = tracks[m2]
--
GitLab