less around the elbow...

scikit-hep · Jan 17, 2024 · 0d89a64 · 0d89a64
1 parent d9d2b6f
commit 0d89a64
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/fastjet/_generalevent.py b/src/fastjet/_generalevent.py
@@ -211,11 +211,10 @@ def _check_subtree_input(self, data):
             return False
 
     def extract_cons(self, array):
-        packed_flattened = ak.flatten(array)
-        px = packed_flattened.px.to_numpy()
-        py = packed_flattened.py.to_numpy()
-        pz = packed_flattened.pz.to_numpy()
-        E = packed_flattened.E.to_numpy()
+        px = ak.to_numpy(array.px.layout.content)
+        py = ak.to_numpy(array.py.layout.content)
+        pz = ak.to_numpy(array.pz.layout.content)
+        E = ak.to_numpy(array.E.layout.content)        
         off = np.asarray(array.layout.stops)
         off = np.insert(off, 0, 0)
         return px, py, pz, E, off