feat(backport): Use 'method' arg for jax.numpy.percentile (#2524)

* Backport PR #2522. * Change the argument name in jax.numpy.percentile from 'interpolation' to 'method'. No change to the lower bounds on jax and jaxlib is required as 'method' already existed as an argument in jax v0.4.1. This just deprecates the use of 'interpolation'. Co-authored-by: Matthew Feickert <[email protected]>
scikit-hep · Jun 20, 2024 · d23f5c4 · d23f5c4
1 parent 47d5930
commit d23f5c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py
@@ -328,7 +328,7 @@ def percentile(self, tensor_in, q, axis=None, interpolation="linear"):
 
         .. versionadded:: 0.7.0
         """
-        return jnp.percentile(tensor_in, q, axis=axis, interpolation=interpolation)
+        return jnp.percentile(tensor_in, q, axis=axis, method=interpolation)
 
     def stack(self, sequence, axis=0):
         return jnp.stack(sequence, axis=axis)