""" This type stub file was generated by pyright. """ import sys from typing import Any, Literal, Protocol, TypeVar, Union from numpy import dtype, float32, float64, int16, int32, int64, int8, uint16, uint32, uint64, uint8 """ This file defines the types for type annotations. These names aren't part of the module namespace, but they are used in the annotations in the function signatures. The functions in the module are only valid for inputs that match the given type annotations. """ __all__ = ["Array", "Device", "Dtype", "SupportsDLPack", "SupportsBufferProtocol", "PyCapsule"] _T_co = TypeVar("_T_co", covariant=True) class NestedSequence(Protocol[_T_co]): def __getitem__(self, key: int, /) -> _T_co | NestedSequence[_T_co]: ... def __len__(self, /) -> int: ... Device = Literal["cpu"] Dtype = dtype[Union[int8, int16, int32, int64, uint8, uint16, uint32, uint64, float32, float64,]] if sys.version_info >= (3, 12): ... else: ... PyCapsule = Any class SupportsDLPack(Protocol): def __dlpack__(self, /, *, stream: None = ...) -> PyCapsule: ...