forgot to_numpy()

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