Add type hints to the numpy.array_api.Array.__array__ signature

Thanks @BvB93

1 files changed, 2 insertions, 1 deletions

diff --git a/numpy/array_api/_array_object.py b/numpy/array_api/_array_object.py
index d322e6ca6..75baf34b0 100644
--- a/numpy/array_api/_array_object.py
+++ b/numpy/array_api/_array_object.py
@@ -33,6 +33,7 @@ from typing import TYPE_CHECKING, Optional, Tuple, Union, Any
 
 if TYPE_CHECKING:
     from ._typing import Any, PyCapsule, Device, Dtype
+    import numpy.typing as npt
 
 import numpy as np
 
@@ -110,7 +111,7 @@ class Array:
 
     # This function is not required by the spec, but we implement it here for
     # convenience so that np.asarray(np.array_api.Array) will work.
-    def __array__(self, dtype=None):
+    def __array__(self, dtype: None | np.dtype[Any] = None) -> npt.NDArray[Any]:
         """
         Warning: this method is NOT part of the array API spec. Implementers
         of other libraries need not include it, and users should not assume it