Merge pull request #20720 from rgommers/arrayapi-annotation-fixes

TYP: add a few type annotations to `numpy.array_api.Array`

Original NumPy Commit: e1123b2d2cf77d123266604328baa3a27c375cf8

Author: mattip

array_api_strict/_array_object.py

@@ -56,6 +56,7 @@ class Array:
     functions, such as asarray().
 
     """
+    _array: np.ndarray
 
     # Use a custom constructor instead of __init__, as manually initializing
     # this class is not supported API.
@@ -125,7 +126,7 @@ def __array__(self, dtype: None | np.dtype[Any] = None) -> npt.NDArray[Any]:
     # spec in places where it either deviates from or is more strict than
     # NumPy behavior
 
-    def _check_allowed_dtypes(self, other, dtype_category, op):
+    def _check_allowed_dtypes(self, other: bool | int | float | Array, dtype_category: str, op: str) -> Array:
         """
         Helper function for operators to only allow specific input dtypes
 
@@ -201,7 +202,7 @@ def _promote_scalar(self, scalar):
         return Array._new(np.array(scalar, self.dtype))
 
     @staticmethod
-    def _normalize_two_args(x1, x2):
+    def _normalize_two_args(x1, x2) -> Tuple[Array, Array]:
         """
         Normalize inputs to two arg functions to fix type promotion rules