1 files changed, 5 insertions, 5 deletions
diff --git a/numpy/_array_api/_searching_functions.py b/numpy/_array_api/_searching_functions.py
index 9a5d583bc..690256430 100644
--- a/numpy/_array_api/_searching_functions.py
+++ b/numpy/_array_api/_searching_functions.py
@@ -4,11 +4,11 @@ from ._array_object import ndarray
 
 from typing import TYPE_CHECKING
 if TYPE_CHECKING:
-    from ._types import Tuple, array
+    from ._types import Tuple, Array
 
 import numpy as np
 
-def argmax(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
+def argmax(x: Array, /, *, axis: int = None, keepdims: bool = False) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.argmax <numpy.argmax>`.
 
@@ -17,7 +17,7 @@ def argmax(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
     # Note: this currently fails as np.argmax does not implement keepdims
     return ndarray._new(np.asarray(np.argmax(x._array, axis=axis, keepdims=keepdims)))
 
-def argmin(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
+def argmin(x: Array, /, *, axis: int = None, keepdims: bool = False) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.argmin <numpy.argmin>`.
 
@@ -26,7 +26,7 @@ def argmin(x: array, /, *, axis: int = None, keepdims: bool = False) -> array:
     # Note: this currently fails as np.argmin does not implement keepdims
     return ndarray._new(np.asarray(np.argmin(x._array, axis=axis, keepdims=keepdims)))
 
-def nonzero(x: array, /) -> Tuple[array, ...]:
+def nonzero(x: Array, /) -> Tuple[Array, ...]:
     """
     Array API compatible wrapper for :py:func:`np.nonzero <numpy.nonzero>`.
 
@@ -34,7 +34,7 @@ def nonzero(x: array, /) -> Tuple[array, ...]:
     """
     return ndarray._new(np.nonzero(x._array))
 
-def where(condition: array, x1: array, x2: array, /) -> array:
+def where(condition: Array, x1: Array, x2: Array, /) -> Array:
     """
     Array API compatible wrapper for :py:func:`np.where <numpy.where>`.