From 5074ce00949244a1e18fe19b25758557c008a1e8 Mon Sep 17 00:00:00 2001 From: Jacan Chaplais Date: Tue, 3 Oct 2023 17:32:01 +0100 Subject: [PATCH] minor internal naming improvement --- graphicle/select.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphicle/select.py b/graphicle/select.py index 1d21292..77868da 100644 --- a/graphicle/select.py +++ b/graphicle/select.py @@ -125,9 +125,10 @@ def fastjet_clusters( sequence = fastjet.ClusterSequence(pmu_ak, jetdef) jets = sequence.inclusive_jets() jet_pmus = gcl.MomentumArray(jets.to_numpy()) + num_jets = len(jet_pmus) pt_descend_idxs = np.argsort(jet_pmus.pt)[::-1].tolist() jet_pmus = jet_pmus[pt_descend_idxs] - cuts = np.ones(len(pt_descend_idxs), dtype=np.bool_) + cuts = np.ones(num_jets, dtype=np.bool_) if pt_cut is not None: cuts = jet_pmus.pt > pt_cut if eta_cut is not None: