hand

venv/lib/python3.12/site-packages/opt_einsum/typing.py

@@ -0,0 +1,27 @@
+"""Types used in the opt_einsum package."""
+
+from collections import namedtuple
+from typing import Any, Callable, Collection, Dict, FrozenSet, List, Literal, Optional, Tuple, Union
+
+TensorShapeType = Tuple[int, ...]
+PathType = Collection[TensorShapeType]
+
+ArrayType = Any
+
+ArrayIndexType = FrozenSet[str]
+ArrayShaped = namedtuple("ArrayShaped", ["shape"])
+
+ContractionListType = List[Tuple[Any, ArrayIndexType, str, Optional[Tuple[str, ...]], Union[str, bool]]]
+PathSearchFunctionType = Callable[[List[ArrayIndexType], ArrayIndexType, Dict[str, int], Optional[int]], PathType]
+
+# Contract kwargs
+OptimizeKind = Union[
+    None,
+    bool,
+    Literal[
+        "optimal", "dp", "greedy", "random-greedy", "random-greedy-128", "branch-all", "branch-2", "auto", "auto-hq"
+    ],
+    PathType,
+    PathSearchFunctionType,
+]
+BackendType = Literal["auto", "object", "autograd", "cupy", "dask", "jax", "theano", "tensorflow", "torch", "libjax"]