39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""
|
|
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:
|
|
...
|
|
|
|
|
|
|