Using typing instead of collections.abc. Fixes #1763

PiperOrigin-RevId: 651186508
Change-Id: I844ec884b802837554b0b60f2414d6bc37f2da14

mjx/mujoco/mjx/_src/collision_convex.py

@@ -14,9 +14,8 @@
 # ==============================================================================
 """Convex collisions."""
 
-from collections.abc import Callable
 import functools
-from typing import Tuple, Union
+from typing import Callable, Tuple, Union
 
 import jax
 from jax import numpy as jp