MAINT: remove local imports from einsum

Author: ev-br

torch_np/_funcs.py

@@ -69,3 +69,4 @@ def __getitem__(self, item):
 s_ = IndexExpression(maketuple=False)
 
 __all__ += ["index_exp", "s_"]
+

torch_np/_funcs_impl.py

@@ -27,6 +27,12 @@
     NotImplementedType,
     OutArray,
     normalize_array_like,
+
+    # these imports are for einsum
+    maybe_copy_to,
+    normalize_casting,
+    normalize_dtype,
+    wrap_tensors,
 )
 
 # ###### array creation routines
@@ -1233,12 +1239,6 @@ def einsum(*operands, out=None, dtype=None, order="K", casting="safe", optimize=
     # Have to manually normalize *operands and **kwargs, following the NumPy signature
 
     from ._ndarray import ndarray
-    from ._normalizations import (
-        maybe_copy_to,
-        normalize_casting,
-        normalize_dtype,
-        wrap_tensors,
-    )
 
     dtype = normalize_dtype(dtype)
     casting = normalize_casting(casting)
@@ -1276,7 +1276,7 @@ def einsum(*operands, out=None, dtype=None, order="K", casting="safe", optimize=
 
     is_short_int = target_dtype in [torch.uint8, torch.int8, torch.int16, torch.int32]
     if is_short_int:
-        target_dtype, result_dtype = torch.int64, target_dtype
+        target_dtype = torch.int64
 
     tensors = _util.typecast_tensors(tensors, target_dtype, casting)