Quick comments and py3.8 fixes

dgasmith · May 20, 2024 · 7e2a610 · 7e2a610
1 parent 807a45d
commit 7e2a610
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/opt_einsum/contract.py b/opt_einsum/contract.py
@@ -143,6 +143,7 @@ def _filter_einsum_defaults(kwargs: Dict[Literal["order", "casting", "dtype", "o
     return ret
 
 
+# Overlaod for contract(str, *operands)
 @overload
 def contract_path(
     subscripts: str,
@@ -155,10 +156,11 @@ def contract_path(
 ) -> Tuple[PathType, PathInfo]: ...
 
 
+# Overlaod for contract(operand, indices, ....)
 @overload
 def contract_path(
     subscripts: ArrayType,
-    *operands: ArrayType | Collection[int],
+    *operands: Union[ArrayType, Collection[int]],
     use_blas: bool = True,
     optimize: OptimizeKind = True,
     memory_limit: _MemoryLimit = None,