
from cupy.cuda import Device as _Device


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

import sys
from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING

from . import Array
from numpy import (
    dtype,
    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 = _Device
if TYPE_CHECKING or sys.version_info >= (3, 9):
    Dtype = dtype[Union[
        int8,
        int16,
        int32,
        int64,
        uint8,
        uint16,
        uint32,
        uint64,
        float32,
        float64,
    ]]
else:
    Dtype = dtype

SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any


from cupy.cuda import Device as _Device


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

import sys
from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING, TypeVar

from . import Array
from numpy import (
    dtype,
    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
_T = TypeVar("_T")
NestedSequence = Sequence[Sequence[_T]]

Device = _Device
if TYPE_CHECKING or sys.version_info >= (3, 9):
    Dtype = dtype[Union[
        int8,
        int16,
        int32,
        int64,
        uint8,
        uint16,
        uint32,
        uint64,
        float32,
        float64,
    ]]
else:
    Dtype = dtype

SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any

Add a missing subscription slot to `NestedSequence`
MAINT: Add a missing subscription slot to `NestedSequence`

  
  from cupy.cuda import Device as _Device
  
  
  __all__ = [
      "Array",
      "Device",
      "Dtype",
      "SupportsDLPack",
      "SupportsBufferProtocol",
      "PyCapsule",
  ]
  
  import sys
- from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING
+ from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING, TypeVar
  
  from . import Array
  from numpy import (
      dtype,
      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
+ _T = TypeVar("_T")
- NestedSequence = Sequence[Sequence[Any]]
+ NestedSequence = Sequence[Sequence[_T]]
  
  Device = _Device
  if TYPE_CHECKING or sys.version_info >= (3, 9):
      Dtype = dtype[Union[
          int8,
          int16,
          int32,
          int64,
          uint8,
          uint16,
          uint32,
          uint64,
          float32,
          float64,
      ]]
  else:
      Dtype = dtype
  
  SupportsDLPack = Any
  SupportsBufferProtocol = Any
  PyCapsule = Any
  
Add a missing subscription slot to `NestedSequence`
## Code Before:

from cupy.cuda import Device as _Device


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

import sys
from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING

from . import Array
from numpy import (
    dtype,
    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 = _Device
if TYPE_CHECKING or sys.version_info >= (3, 9):
    Dtype = dtype[Union[
        int8,
        int16,
        int32,
        int64,
        uint8,
        uint16,
        uint32,
        uint64,
        float32,
        float64,
    ]]
else:
    Dtype = dtype

SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any

## Instruction:
Add a missing subscription slot to `NestedSequence`
## Code After:

from cupy.cuda import Device as _Device


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

import sys
from typing import Any, Literal, Sequence, Type, Union, TYPE_CHECKING, TypeVar

from . import Array
from numpy import (
    dtype,
    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
_T = TypeVar("_T")
NestedSequence = Sequence[Sequence[_T]]

Device = _Device
if TYPE_CHECKING or sys.version_info >= (3, 9):
    Dtype = dtype[Union[
        int8,
        int16,
        int32,
        int64,
        uint8,
        uint16,
        uint32,
        uint64,
        float32,
        float64,
    ]]
else:
    Dtype = dtype

SupportsDLPack = Any
SupportsBufferProtocol = Any
PyCapsule = Any
