From d7a151748e221509d4abf56dda98b53368eab7fa Mon Sep 17 00:00:00 2001 From: Wessel Bruinsma Date: Fri, 20 May 2022 10:12:03 +0200 Subject: [PATCH] Fix `quantile` --- lab/numpy/generic.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lab/numpy/generic.py b/lab/numpy/generic.py index 42dfa10..d28410a 100644 --- a/lab/numpy/generic.py +++ b/lab/numpy/generic.py @@ -331,4 +331,8 @@ def argsort(a: Numeric, axis: Int = -1, descending: bool = False): @dispatch def quantile(a: Numeric, q: Numeric, axis: Union[Int, None] = None): - return np.quantile(a, q, axis=axis, method="linear") + if tuple(map(int, np.__version__.split("."))) >= (1, 22): # pragma: no cover + method = {"method": "linear"} + else: # pragma: no cover + method = {"interpolation": "linear"} + return np.quantile(a, q, axis=axis, **method)