Remove npy<2 compatibility for numpy_maxdims

Author: Armavica

pytensor/npy_2_compat.py

@@ -24,10 +24,6 @@
     numpy_axis_is_none_flag = 32
 
 
-# max number of dims is 64 in numpy 2.x; 32 in older versions
-numpy_maxdims = 64 if using_numpy_2 else 32
-
-
 # function that replicates np.unique from numpy < 2.0
 def old_np_unique(
     arr, return_index=False, return_inverse=False, return_counts=False, axis=None

tests/tensor/test_elemwise.py

@@ -19,7 +19,6 @@
 from pytensor.graph.replace import vectorize_node
 from pytensor.link.basic import PerformLinker
 from pytensor.link.c.basic import CLinker, OpWiseCLinker
-from pytensor.npy_2_compat import numpy_maxdims
 from pytensor.scalar import ScalarOp, float32, float64, int32, int64
 from pytensor.tensor import as_tensor_variable
 from pytensor.tensor.basic import get_scalar_constant_value, second
@@ -168,6 +167,7 @@ def test_infer_shape(self):
             )
 
     def test_too_big_rank(self):
+        numpy_maxdims = 64
         x = self.type(self.dtype, shape=())()
         y = x.dimshuffle(("x",) * (numpy_maxdims + 1))