
__all__ = ['Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
           'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']

from typing import Literal, Optional, Tuple, Union, TypeVar

import numpy as np

array = np.ndarray
device = TypeVar('device')
dtype = Literal[np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16,
                np.uint32, np.uint64, np.float32, np.float64]
SupportsDLPack = TypeVar('SupportsDLPack')
SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
PyCapsule = TypeVar('PyCapsule')


__all__ = ['Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
           'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']

from typing import Literal, Optional, Tuple, Union, TypeVar

from . import (ndarray, 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')

Use the array API types for the array API type annotations
Use the array API types for the array API type annotations

  
  __all__ = ['Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
             'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']
  
  from typing import Literal, Optional, Tuple, Union, TypeVar
  
- import numpy as np
+ from . import (ndarray, int8, int16, int32, int64, uint8, uint16, uint32,
+                uint64, float32, float64)
  
- array = np.ndarray
+ array = ndarray
  device = TypeVar('device')
- dtype = Literal[np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16,
+ dtype = Literal[int8, int16, int32, int64, uint8, uint16,
-                 np.uint32, np.uint64, np.float32, np.float64]
+                 uint32, uint64, float32, float64]
  SupportsDLPack = TypeVar('SupportsDLPack')
  SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
  PyCapsule = TypeVar('PyCapsule')
  
Use the array API types for the array API type annotations
## Code Before:

__all__ = ['Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
           'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']

from typing import Literal, Optional, Tuple, Union, TypeVar

import numpy as np

array = np.ndarray
device = TypeVar('device')
dtype = Literal[np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16,
                np.uint32, np.uint64, np.float32, np.float64]
SupportsDLPack = TypeVar('SupportsDLPack')
SupportsBufferProtocol = TypeVar('SupportsBufferProtocol')
PyCapsule = TypeVar('PyCapsule')

## Instruction:
Use the array API types for the array API type annotations
## Code After:

__all__ = ['Literal', 'Optional', 'Tuple', 'Union', 'array', 'device',
           'dtype', 'SupportsDLPack', 'SupportsBufferProtocol', 'PyCapsule']

from typing import Literal, Optional, Tuple, Union, TypeVar

from . import (ndarray, 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')
