jax.numpy.clip: update use of deprecated arguments.

- a is now positional-only
- a_min is now min
- a_max is now max

The old argument names have been deprecated since JAX v0.4.27.

PiperOrigin-RevId: 714268837

Author: Jake VanderPlas

flow_field.py

@@ -144,9 +144,9 @@ def fmax(x, v=0.0):
     out[nonzero_indices] = xcorr[nonzero_indices] / denom[nonzero_indices]
 
   if use_jax:
-    out = jnp.clip(out, a_min=-1, a_max=1)
+    out = jnp.clip(out, min=-1, max=1)
   else:
-    np.clip(out, a_min=-1, a_max=1, out=out)
+    np.clip(out, min=-1, max=1, out=out)
 
   px_threshold = 0.3 * xnp.max(overlap_masked_px, keepdims=True)
   if use_jax: