returning npt.ArrayLike instead of NDArrayLike because of scalar return values

Author: brokkoli71

src/zarr/api/asynchronous.py

@@ -232,7 +232,7 @@ async def load(
     path: str | None = None,
     zarr_format: ZarrFormat | None = None,
     zarr_version: ZarrFormat | None = None,
-) -> NDArrayLike | dict[str, NDArrayLike]:
+) -> npt.ArrayLike | dict[str, npt.ArrayLike]:
     """Load data from an array or group into memory.
 
     Parameters

src/zarr/core/array.py

@@ -1204,7 +1204,7 @@ async def _get_selection(
         prototype: BufferPrototype,
         out: NDBuffer | None = None,
         fields: Fields | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         # check fields are sensible
         out_dtype = check_fields(fields, self.dtype)
 
@@ -1254,7 +1254,7 @@ async def getitem(
         selection: BasicSelection,
         *,
         prototype: BufferPrototype | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """
         Asynchronous function that retrieves a subset of the array's data based on the provided selection.
 
@@ -1267,7 +1267,7 @@ async def getitem(
 
         Returns
         -------
-        NDArrayLike
+        npt.ArrayLike
             The retrieved subset of the array's data.
 
         Examples
@@ -2225,7 +2225,7 @@ def __array__(
 
         return arr_np
 
-    def __getitem__(self, selection: Selection) -> NDArrayLike:
+    def __getitem__(self, selection: Selection) -> npt.ArrayLike:
         """Retrieve data for an item or region of the array.
 
         Parameters
@@ -2236,7 +2236,7 @@ def __getitem__(self, selection: Selection) -> NDArrayLike:
 
         Returns
         -------
-        NDArrayLike
+        npt.ArrayLike
              An array-like containing the data for the requested region.
 
         Examples
@@ -2483,7 +2483,7 @@ def get_basic_selection(
         out: NDBuffer | None = None,
         prototype: BufferPrototype | None = None,
         fields: Fields | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """Retrieve data for an item or region of the array.
 
         Parameters
@@ -2703,7 +2703,7 @@ def get_orthogonal_selection(
         out: NDBuffer | None = None,
         fields: Fields | None = None,
         prototype: BufferPrototype | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """Retrieve data by making a selection for each dimension of the array. For
         example, if an array has 2 dimensions, allows selecting specific rows and/or
         columns. The selection for each dimension can be either an integer (indexing a
@@ -2939,7 +2939,7 @@ def get_mask_selection(
         out: NDBuffer | None = None,
         fields: Fields | None = None,
         prototype: BufferPrototype | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """Retrieve a selection of individual items, by providing a Boolean array of the
         same shape as the array against which the selection is being made, where True
         values indicate a selected item.
@@ -3101,7 +3101,7 @@ def get_coordinate_selection(
         out: NDBuffer | None = None,
         fields: Fields | None = None,
         prototype: BufferPrototype | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """Retrieve a selection of individual items, by providing the indices
         (coordinates) for each selected item.
 
@@ -3289,7 +3289,7 @@ def get_block_selection(
         out: NDBuffer | None = None,
         fields: Fields | None = None,
         prototype: BufferPrototype | None = None,
-    ) -> NDArrayLike:
+    ) -> npt.ArrayLike:
         """Retrieve a selection of individual items, by providing the indices
         (coordinates) for each selected item.
 
@@ -3307,7 +3307,7 @@ def get_block_selection(
 
         Returns
         -------
-        NDArrayLike
+        npt.ArrayLike
             An array-like containing the data for the requested block selection.
 
         Examples

src/zarr/core/indexing.py

@@ -929,7 +929,7 @@ class OIndex:
     array: Array
 
     # TODO: develop Array generic and move zarr.Array[np.intp] | zarr.Array[np.bool_] to ArrayOfIntOrBool
-    def __getitem__(self, selection: OrthogonalSelection | Array) -> NDArrayLike:
+    def __getitem__(self, selection: OrthogonalSelection | Array) -> npt.ArrayLike:
         from zarr.core.array import Array
 
         # if input is a Zarr array, we materialize it now.
@@ -1038,7 +1038,7 @@ def __iter__(self) -> Iterator[ChunkProjection]:
 class BlockIndex:
     array: Array
 
-    def __getitem__(self, selection: BasicSelection) -> NDArrayLike:
+    def __getitem__(self, selection: BasicSelection) -> npt.ArrayLike:
         fields, new_selection = pop_fields(selection)
         new_selection = ensure_tuple(new_selection)
         new_selection = replace_lists(new_selection)
@@ -1227,7 +1227,7 @@ class VIndex:
     array: Array
 
     # TODO: develop Array generic and move zarr.Array[np.intp] | zarr.Array[np.bool_] to ArrayOfIntOrBool
-    def __getitem__(self, selection: CoordinateSelection | MaskSelection | Array) -> NDArrayLike:
+    def __getitem__(self, selection: CoordinateSelection | MaskSelection | Array) -> npt.ArrayLike:
         from zarr.core.array import Array
 
         # if input is a Zarr array, we materialize it now.