use ak._slicing.normalize_integer_like(head)

scikit-hep · Sep 19, 2024 · bd7b125 · bd7b125
1 parent ed55752
commit bd7b125
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/awkward/contents/listarray.py b/src/awkward/contents/listarray.py
@@ -712,6 +712,7 @@ def _getitem_next(
             nexthead, nexttail = ak._slicing.head_tail(tail)
             lenstarts = self._starts.length
             nextcarry = ak.index.Index64.empty(lenstarts, self._backend.index_nplike)
+            head = ak._slicing.normalize_integer_like(head)
             assert (
                 nextcarry.nplike is self._backend.index_nplike
                 and self._starts.nplike is self._backend.index_nplike
@@ -728,9 +729,7 @@ def _getitem_next(
                     self._starts.data,
                     self._stops.data,
                     lenstarts,
-                    int(
-                        head
-                    ),  # NOTE: when 'at' is an integer it is always 'int64' for both CPU and GPU kernels
+                    head,
                 ),
                 slicer=head,
             )