summaryrefslogtreecommitdiff
path: root/numpy/typing/_shape.py
blob: 75698f3d332c20b126ece5de9b6e83d8beeadc06 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import sys
from typing import Sequence, Tuple, Union, Any

from . import _HAS_TYPING_EXTENSIONS

if sys.version_info >= (3, 8):
    from typing import SupportsIndex
elif _HAS_TYPING_EXTENSIONS:
    from typing_extensions import SupportsIndex
else:
    SupportsIndex = Any

_Shape = Tuple[int, ...]

# Anything that can be coerced to a shape tuple
_ShapeLike = Union[SupportsIndex, Sequence[SupportsIndex]]