unit tests working locally

HenryDayHall · Apr 8, 2023 · 497b528 · 497b528
1 parent ed800ca
commit 497b528
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/jet_tools/MassPeaks.py b/jet_tools/MassPeaks.py
@@ -65,10 +65,8 @@ def combined_jet_mass(eventWise, jet_name, jet_idxs,
 
     """
     assert eventWise.selected_event is not None
-    try:
-        jet_idxs = list(jet_idxs)
-    except TypeError:
-        pass
+    if hasattr(jet_idxs, '__len__') and not len(jet_idxs):
+        return 0.
     if signal_background == 'both':  # jsut use the jet roots
         jet_roots = getattr(eventWise, jet_name + "_Parent") == -1
         e = ak.flatten(getattr(eventWise, jet_name + "_Energy")[jet_roots])[jet_idxs]