typing.py

from typing import Any, Tuple, Type, Union

__all__ = [
    "DataType",
    "Scalar",
    "ScalarType",
    "Array",
    "Shape",
    "AtomicIndex",
    "Index",
    "Param",
]

DataType = Type[Any]
Scalar = Union[bool, int, float, complex]
ScalarType = Union[Type[bool], Type[int], Type[float], Type[complex]]
Array = Any
Shape = Tuple[int, ...]
AtomicIndex = Union[int, "ellipsis", slice, None]  # noqa
Index = Union[AtomicIndex, Tuple[AtomicIndex, ...]]
Param = Tuple