summaryrefslogtreecommitdiff
path: root/numpy/typing/_shape.py
blob: 0742be8a95d8a883fd65047be6411dc16ee689d4 (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

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 = NotImplemented

_Shape = Tuple[int, ...]

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