From f498a8eec0324d04c6adfd99f040cf27305e5ae4 Mon Sep 17 00:00:00 2001 From: dachengx Date: Wed, 19 Mar 2025 00:25:30 -0500 Subject: [PATCH] Add arguments of `numpy.argsort`, like `order` --- strax/sort_enforcement.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/strax/sort_enforcement.py b/strax/sort_enforcement.py index 84bda23da..fd285ea58 100644 --- a/strax/sort_enforcement.py +++ b/strax/sort_enforcement.py @@ -30,7 +30,7 @@ def stable_sort(arr, kind="mergesort", **kwargs): @register_jitable -def stable_argsort(arr, kind="mergesort"): +def stable_argsort(arr, kind="mergesort", **kwargs): """Numba-optimized stable argsort function using mergesort. Args: @@ -43,4 +43,4 @@ def stable_argsort(arr, kind="mergesort"): """ if kind != "mergesort": raise SortingError(UNSTABLE_SORT_MESSAGE) - return np.argsort(arr, kind="mergesort") + return np.argsort(arr, kind="mergesort", **kwargs)