Backport PR #2522: feat: Use 'method' arg for jax.numpy.percentile

scikit-hep · Jun 20, 2024 · ce1753b · ce1753b
1 parent 47d5930
commit ce1753b
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)