Remove npy<2 compatibility for einsum helpers

Author: Armavica

pytensor/npy_2_compat.py

@@ -3,25 +3,6 @@
 import numpy as np
 
 
-try:
-    from numpy._core.einsumfunc import (  # type: ignore[attr-defined]
-        _find_contraction,
-        _parse_einsum_input,
-    )
-except ModuleNotFoundError:
-    from numpy.core.einsumfunc import (  # type: ignore[no-redef]
-        _find_contraction,
-        _parse_einsum_input,
-    )
-
-
-# suppress linting warning by "using" the imports here:
-__all__ = [
-    "_find_contraction",
-    "_parse_einsum_input",
-]
-
-
 numpy_version_tuple = tuple(int(n) for n in np.__version__.split(".")[:2])
 numpy_version = np.lib.NumpyVersion(
     np.__version__

pytensor/tensor/einsum.py

@@ -6,10 +6,13 @@
 from typing import cast
 
 import numpy as np
+from numpy._core.einsumfunc import (  # type: ignore[attr-defined]
+    _find_contraction,
+    _parse_einsum_input,
+)
 from numpy.lib.array_utils import normalize_axis_index, normalize_axis_tuple
 
 from pytensor.compile.builders import OpFromGraph
-from pytensor.npy_2_compat import _find_contraction, _parse_einsum_input
 from pytensor.tensor import TensorLike
 from pytensor.tensor.basic import (
     arange,