From 37b0d2185ff6aa9a2d141fcca78db56a794cdc8f Mon Sep 17 00:00:00 2001 From: benjamindonnachie <83379521+benjamindonnachie@users.noreply.github.com> Date: Sat, 17 Aug 2024 16:02:00 +0100 Subject: [PATCH 1/2] Update lib.pyi Update maybe_indices_to_slice to use np.intp for max_len. --- pandas/_libs/lib.pyi | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/_libs/lib.pyi b/pandas/_libs/lib.pyi index daaaacee3487d..ebdf78d04379b 100644 --- a/pandas/_libs/lib.pyi +++ b/pandas/_libs/lib.pyi @@ -161,7 +161,7 @@ def maybe_booleans_to_slice( ) -> slice | npt.NDArray[np.uint8]: ... def maybe_indices_to_slice( indices: npt.NDArray[np.intp], - max_len: int, + max_len: np.intp, ) -> slice | npt.NDArray[np.intp]: ... def is_all_arraylike(obj: list) -> bool: ... From 25e754a815f277dcdca11c668c26999a566204c8 Mon Sep 17 00:00:00 2001 From: benjamindonnachie <83379521+benjamindonnachie@users.noreply.github.com> Date: Sat, 17 Aug 2024 16:03:37 +0100 Subject: [PATCH 2/2] Update lib.pyx Update maybe_indices_to_slice to use intp_t for max_len rather than int --- pandas/_libs/lib.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pandas/_libs/lib.pyx b/pandas/_libs/lib.pyx index 489d4fa111d40..75ab3259ccef6 100644 --- a/pandas/_libs/lib.pyx +++ b/pandas/_libs/lib.pyx @@ -494,7 +494,7 @@ def has_only_ints_or_nan(const floating[:] arr) -> bool: return True -def maybe_indices_to_slice(ndarray[intp_t, ndim=1] indices, int max_len): +def maybe_indices_to_slice(ndarray[intp_t, ndim=1] indices, intp_t max_len): cdef: Py_ssize_t i, n = len(indices) intp_t k, vstart, vlast, v