Merge branch 'add_array_api_mean' of https://github.com/Illviljan/xarray

into add_array_api_mean
pydata · Aug 19, 2024 · ab828a2 · ab828a2
2 parents a0489f0 + 8098fa3
commit ab828a2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/xarray/namedarray/_array_api/_creation_functions.py b/xarray/namedarray/_array_api/_creation_functions.py
@@ -5,8 +5,8 @@
 import numpy as np
 
 from xarray.namedarray._array_api._utils import (
-    _maybe_default_namespace,
     _get_data_namespace,
+    _maybe_default_namespace,
 )
 from xarray.namedarray._typing import (
     Default,

diff --git a/xarray/namedarray/_array_api/_info.py b/xarray/namedarray/_array_api/_info.py
@@ -4,11 +4,16 @@
 
 if TYPE_CHECKING:
     from types import ModuleType
-    from typing import Optional, Union, Tuple, List
+    from typing import Optional, Union
+
     from xarray.namedarray._typing import _Device
 
 from xarray.namedarray._array_api._dtypes import (
     bool,
+    complex64,
+    complex128,
+    float32,
+    float64,
     int8,
     int16,
     int32,
@@ -17,10 +22,6 @@
     uint16,
     uint32,
     uint64,
-    float32,
-    float64,
-    complex64,
-    complex128,
 )
 
 
@@ -60,7 +61,7 @@ def default_dtypes(
 def dtypes(
     *,
     device: _Device | None = None,
-    kind: Union[str, Tuple[str, ...]] | None = None,
+    kind: str| Tuple[str, ...] | None = None,
 ) -> dict:
     if kind is None:
         return {
@@ -138,7 +139,7 @@ def dtypes(
     raise ValueError(f"unsupported kind: {kind!r}")
 
 
-def devices() -> List[_Device]:
+def devices() -> list[_Device]:
     return [default_device()]
 
 

diff --git a/xarray/namedarray/_array_api/_manipulation_functions.py b/xarray/namedarray/_array_api/_manipulation_functions.py
@@ -2,9 +2,8 @@
 
 from typing import Any
 
-from xarray.namedarray._array_api._utils import _get_data_namespace
 from xarray.namedarray._array_api._creation_functions import asarray
-
+from xarray.namedarray._array_api._utils import _get_data_namespace
 from xarray.namedarray._typing import (
     Default,
     _arrayapi,