Add defaults for min_value and max_value to tree_utils.tree_clip.

Author: carlosgmartin

optax/tree_utils/_tree_math.py

@@ -323,8 +323,8 @@ def tree_full_like(
 
 def tree_clip(
     tree: Any,
-    min_value: Optional[jax.typing.ArrayLike],
-    max_value: Optional[jax.typing.ArrayLike],
+    min_value: Optional[jax.typing.ArrayLike] = None,
+    max_value: Optional[jax.typing.ArrayLike] = None,
 ) -> Any:
   """Creates an identical tree where all tensors are clipped to `[min, max]`.