Update npy_2_compat imports

Author: jessegrabowski

pymc/distributions/transforms.py

@@ -17,8 +17,8 @@
 import numpy as np
 import pytensor.tensor as pt
 
+from numpy.lib.array_utils import normalize_axis_tuple
 from pytensor.graph import Op
-from pytensor.npy_2_compat import normalize_axis_tuple
 from pytensor.tensor import TensorVariable
 
 from pymc.logprob.transforms import (

pymc/logprob/tensor.py

@@ -37,10 +37,10 @@
 
 from pathlib import Path
 
+from numpy.lib.array_utils import normalize_axis_index
 from pytensor import tensor as pt
 from pytensor.graph.fg import FunctionGraph
 from pytensor.graph.rewriting.basic import node_rewriter
-from pytensor.npy_2_compat import normalize_axis_index
 from pytensor.tensor import TensorVariable
 from pytensor.tensor.basic import Join, MakeVector, Split
 from pytensor.tensor.elemwise import DimShuffle, Elemwise