
__all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
           'SupportsBufferProtocol', 'PyCapsule']

from typing import Any, Sequence, Type, Union

from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
               uint64, float32, float64)

Array = ndarray
Device = TypeVar('device')
Dtype = Literal[int8, int16, int32, int64, uint8, uint16,
                uint32, uint64, float32, float64]
SupportsDLPack = TypeVar('SupportsDLPack')
SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
PyCapsule = TypeVar('PyCapsule')


__all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
           'SupportsBufferProtocol', 'PyCapsule']

from typing import Any, Sequence, Type, Union

from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
               uint64, float32, float64)

# This should really be recursive, but that isn't supported yet. See the
# similar comment in numpy/typing/_array_like.py
NestedSequence = Sequence[Sequence[Any]]

Device = Any
Dtype = Type[Union[[int8, int16, int32, int64, uint8, uint16,
                   uint32, uint64, float32, float64]]]
SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any

Use better type definitions for the array API custom types
Use better type definitions for the array API custom types

  
  __all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
             'SupportsBufferProtocol', 'PyCapsule']
  
  from typing import Any, Sequence, Type, Union
  
  from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
                 uint64, float32, float64)
  
+ # This should really be recursive, but that isn't supported yet. See the
+ # similar comment in numpy/typing/_array_like.py
+ NestedSequence = Sequence[Sequence[Any]]
- Array = ndarray
- Device = TypeVar('device')
- Dtype = Literal[int8, int16, int32, int64, uint8, uint16,
-                 uint32, uint64, float32, float64]
- SupportsDLPack = TypeVar('SupportsDLPack')
- SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
- PyCapsule = TypeVar('PyCapsule')
  
+ Device = Any
+ Dtype = Type[Union[[int8, int16, int32, int64, uint8, uint16,
+                    uint32, uint64, float32, float64]]]
+ SupportsDLPack = Any
+ SupportsBufferProtocol = Any
+ PyCapsule = Any
+ 
Use better type definitions for the array API custom types
## Code Before:

__all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
           'SupportsBufferProtocol', 'PyCapsule']

from typing import Any, Sequence, Type, Union

from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
               uint64, float32, float64)

Array = ndarray
Device = TypeVar('device')
Dtype = Literal[int8, int16, int32, int64, uint8, uint16,
                uint32, uint64, float32, float64]
SupportsDLPack = TypeVar('SupportsDLPack')
SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
PyCapsule = TypeVar('PyCapsule')

## Instruction:
Use better type definitions for the array API custom types
## Code After:

__all__ = ['Array', 'Device', 'Dtype', 'SupportsDLPack',
           'SupportsBufferProtocol', 'PyCapsule']

from typing import Any, Sequence, Type, Union

from . import (Array, int8, int16, int32, int64, uint8, uint16, uint32,
               uint64, float32, float64)

# This should really be recursive, but that isn't supported yet. See the
# similar comment in numpy/typing/_array_like.py
NestedSequence = Sequence[Sequence[Any]]

Device = Any
Dtype = Type[Union[[int8, int16, int32, int64, uint8, uint16,
                   uint32, uint64, float32, float64]]]
SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any
