Compatible normalize_axis_tuple import

Author: ricardoV94

pymc/distributions/transforms.py

@@ -18,11 +18,8 @@
 import numpy as np
 import pytensor.tensor as pt
 
-
-# ignore mypy error because it somehow considers that
-# "numpy.core.numeric has no attribute normalize_axis_tuple"
-from numpy.core.numeric import normalize_axis_tuple  # type: ignore[attr-defined]
 from pytensor.graph import Op
+from pytensor.npy_2_compat import normalize_axis_tuple
 from pytensor.tensor import TensorVariable
 
 from pymc.logprob.transforms import (