Dtype -> DType

Author: crusaderky

array_api_compat/common/_aliases.py

@@ -8,7 +8,7 @@
 from typing import Any, NamedTuple, Optional, Sequence, Tuple, Union
 
 from ._helpers import array_namespace, _check_device, device, is_torch_array, is_cupy_namespace
-from ._typing import Array, Device, Dtype, Namespace
+from ._typing import Array, Device, DType, Namespace
 
 # These functions are modified from the NumPy versions.
 
@@ -21,7 +21,7 @@ def arange(
     step: Union[int, float] = 1,
     *,
     xp: Namespace,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -32,7 +32,7 @@ def empty(
     shape: Union[int, Tuple[int, ...]],
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -44,7 +44,7 @@ def empty_like(
     /,
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None, 
+    dtype: Optional[DType] = None, 
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -58,7 +58,7 @@ def eye(
     *,
     xp: Namespace,
     k: int = 0,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -70,7 +70,7 @@ def full(
     fill_value: Union[int, float],
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -83,7 +83,7 @@ def full_like(
     fill_value: Union[int, float],
     *,
     xp: Namespace,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -97,7 +97,7 @@ def linspace(
     num: int,
     *,
     xp: Namespace,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     endpoint: bool = True,
     **kwargs,
@@ -109,7 +109,7 @@ def ones(
     shape: Union[int, Tuple[int, ...]],
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -121,7 +121,7 @@ def ones_like(
     /,
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -132,7 +132,7 @@ def zeros(
     shape: Union[int, Tuple[int, ...]],
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -144,7 +144,7 @@ def zeros_like(
     /,
     xp: Namespace,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -279,7 +279,7 @@ def cumulative_sum(
     xp: Namespace,
     *,
     axis: Optional[int] = None,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     include_initial: bool = False,
     **kwargs,
 ) -> Array:
@@ -310,7 +310,7 @@ def cumulative_prod(
     xp: Namespace,
     *,
     axis: Optional[int] = None,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     include_initial: bool = False,
     **kwargs,
 ) -> Array:
@@ -537,8 +537,8 @@ def vecdot(x1: Array, x2: Array, /, xp: Namespace, *, axis: int = -1) -> Array:
 # isdtype is a new function in the 2022.12 array API specification.
 
 def isdtype(
-    dtype: Dtype,
-    kind: Union[Dtype, str, Tuple[Union[Dtype, str], ...]],
+    dtype: DType,
+    kind: Union[DType, str, Tuple[Union[DType, str], ...]],
     xp: Namespace,
     *,
     _tuple: bool = True, # Disallow nested tuples

array_api_compat/common/_fft.py

@@ -3,7 +3,7 @@
 from collections.abc import Sequence
 from typing import Union, Optional, Literal
 
-from ._typing import Device, Array, Dtype, Namespace
+from ._typing import Device, Array, DType, Namespace
 
 # Note: NumPy fft functions improperly upcast float32 and complex64 to
 # complex128, which is why we require wrapping them all here.
@@ -154,7 +154,7 @@ def fftfreq(
     xp: Namespace,
     *,
     d: float = 1.0,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
 ) -> Array:
     if device not in ["cpu", None]:
@@ -170,7 +170,7 @@ def rfftfreq(
     xp: Namespace,
     *,
     d: float = 1.0,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
 ) -> Array:
     if device not in ["cpu", None]:

array_api_compat/common/_typing.py

@@ -4,8 +4,8 @@
 
 __all__ = [
     "Array",
+    "DType",
     "Device",
-    "Dtype",
     "Namespace",
     "NestedSequence",
     "SupportsBufferProtocol",
@@ -22,5 +22,5 @@ def __len__(self, /) -> int: ...
 
 Array = Any
 Device = Any
-Dtype = Any
+DType = Any
 Namespace = ModuleType

array_api_compat/cupy/_aliases.py

@@ -8,7 +8,7 @@
 from ..common._typing import NestedSequence, SupportsBufferProtocol
 from .._internal import get_xp
 from ._info import __array_namespace_info__
-from ._typing import Array, Device, Dtype
+from ._typing import Array, Device, DType
 
 bool = cp.bool_
 
@@ -77,7 +77,7 @@ def asarray(
     ],
     /,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     copy: Optional[bool] = _copy_default,
     **kwargs,
@@ -112,7 +112,7 @@ def asarray(
 
 def astype(
     x: Array,
-    dtype: Dtype,
+    dtype: DType,
     /,
     *,
     copy: bool = True,

array_api_compat/cupy/_typing.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-__all__ = ["Array", "Device", "Dtype"]
+__all__ = ["Array", "DType", "Device"]
 _all_ignore = ["cp"]
 
 from typing import Union
@@ -10,7 +10,7 @@
 from cupy.cuda.device import Device
 
 try:
-    Dtype = cp.dtype[
+    DType = cp.dtype[
         Union[
             cp.intp,
             cp.int8,
@@ -30,4 +30,4 @@
     ]
 except TypeError:
     # NumPy 1.x on Python 3.9 and 3.10
-    Dtype = cp.dtype
+    DType = cp.dtype

array_api_compat/dask/array/_aliases.py

@@ -4,7 +4,7 @@
 
 import numpy as np
 from numpy import (
-    # Dtypes
+    # dtypes
     iinfo,
     finfo,
     bool_ as bool,
@@ -29,7 +29,7 @@
 from ...common._typing import (
     Array,
     Device,
-    Dtype,
+    DType,
     NestedSequence,
     SupportsBufferProtocol,
 )
@@ -43,7 +43,7 @@
 # da.astype doesn't respect copy=True
 def astype(
     x: Array,
-    dtype: Dtype,
+    dtype: DType,
     /,
     *,
     copy: bool = True,
@@ -75,7 +75,7 @@ def arange(
     stop: Optional[Union[int, float]] = None,
     step: Union[int, float] = 1,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     **kwargs,
 ) -> Array:
@@ -146,7 +146,7 @@ def asarray(
     ],
     /,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     copy: Optional[Union[bool, np._CopyMode]] = None,
     **kwargs,

array_api_compat/numpy/_aliases.py

@@ -6,7 +6,7 @@
 from ..common import _aliases
 from ..common._typing import NestedSequence, SupportsBufferProtocol
 from ._info import __array_namespace_info__
-from ._typing import Array, Device, Dtype
+from ._typing import Array, Device, DType
 
 import numpy as np
 bool = np.bool_
@@ -84,7 +84,7 @@ def asarray(
     ],
     /,
     *,
-    dtype: Optional[Dtype] = None,
+    dtype: Optional[DType] = None,
     device: Optional[Device] = None,
     copy: "Optional[Union[bool, np._CopyMode]]" = None,
     **kwargs,
@@ -116,7 +116,7 @@ def asarray(
 
 def astype(
     x: Array,
-    dtype: Dtype,
+    dtype: DType,
     /,
     *,
     copy: bool = True,

array_api_compat/numpy/_typing.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-__all__ = ["Array", "Device", "Dtype"]
+__all__ = ["Array", "DType", "Device"]
 _all_ignore = ["np"]
 
 from typing import Literal, Union
@@ -10,7 +10,7 @@
 
 Device = Literal["cpu"]
 try:
-    Dtype = np.dtype[
+    DType = np.dtype[
         Union[
             np.intp,
             np.int8,
@@ -30,4 +30,4 @@
     ]
 except TypeError:
     # NumPy 1.x on Python 3.9 and 3.10
-    Dtype = np.dtype
+    DType = np.dtype