use general Array protocol for other in comparisons

Signed-off-by: nstarman <[email protected]>

src/_array_api_conf.py

@@ -66,6 +66,7 @@
 ]
 nitpick_ignore_regex = [
     ("py:class", ".*array"),
+    ("py:class", ".*Array"),
     ("py:class", ".*device"),
     ("py:class", ".*Device"),
     ("py:class", ".*dtype"),

src/array_api_stubs/_draft/array_object.py

@@ -159,7 +159,7 @@ def __abs__(self: array, /) -> array:
         """
         ...
 
-    def __add__(self: array, other: int | float | array, /) -> array:
+    def __add__(self: array, other: int | float | Array, /) -> array:
         """
         Calculates the sum for each element of an array instance with the respective element of the array ``other``.
 
@@ -513,7 +513,7 @@ def __dlpack_device__(self, /) -> tuple[Enum, int]:
     # Note that __eq__ returns an array while `object.__eq__` returns a bool.
     # Hence Mypy will complain that this violates the Liskov substitution
     # principle - ignore that.
-    def __eq__(self: array, other: int | float | bool | array, /) -> array:  # type: ignore[override]
+    def __eq__(self: array, other: int | float | bool | Array, /) -> array:  # type: ignore[override]
         r"""
         Computes the truth value of ``self_i == other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -577,7 +577,7 @@ def __float__(self, /) -> float:
         """
         ...
 
-    def __floordiv__(self: array, other: int | float | array, /) -> array:
+    def __floordiv__(self: array, other: int | float | Array, /) -> array:
         """
         Evaluates ``self_i // other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -602,7 +602,7 @@ def __floordiv__(self: array, other: int | float | array, /) -> array:
         """
         ...
 
-    def __ge__(self: array, other: int | float | array, /) -> array:
+    def __ge__(self: array, other: int | float | Array, /) -> array:
         """
         Computes the truth value of ``self_i >= other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -658,7 +658,7 @@ def __getitem__(
         """
         ...
 
-    def __gt__(self: array, other: int | float | array, /) -> array:
+    def __gt__(self: array, other: int | float | Array, /) -> array:
         """
         Computes the truth value of ``self_i > other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -786,7 +786,7 @@ def __invert__(self: array, /) -> array:
         """
         ...
 
-    def __le__(self: array, other: int | float | array, /) -> array:
+    def __le__(self: array, other: int | float | Array, /) -> array:
         """
         Computes the truth value of ``self_i <= other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -836,7 +836,7 @@ def __lshift__(self: array, other: int | array, /) -> array:
         """
         ...
 
-    def __lt__(self: array, other: int | float | array, /) -> array:
+    def __lt__(self: array, other: int | float | Array, /) -> array:
         """
         Computes the truth value of ``self_i < other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -913,7 +913,7 @@ def __matmul__(self: array, other: array, /) -> array:
         """
         ...
 
-    def __mod__(self: array, other: int | float | array, /) -> array:
+    def __mod__(self: array, other: int | float | Array, /) -> array:
         """
         Evaluates ``self_i % other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -938,7 +938,7 @@ def __mod__(self: array, other: int | float | array, /) -> array:
         """
         ...
 
-    def __mul__(self: array, other: int | float | array, /) -> array:
+    def __mul__(self: array, other: int | float | Array, /) -> array:
         r"""
         Calculates the product for each element of an array instance with the respective element of the array ``other``.
 
@@ -969,7 +969,7 @@ def __mul__(self: array, other: int | float | array, /) -> array:
         ...
 
     # See note above __eq__ method for explanation of the `type: ignore`
-    def __ne__(self: array, other: int | float | bool | array, /) -> array:  # type: ignore[override]
+    def __ne__(self: array, other: int | float | bool | Array, /) -> array:  # type: ignore[override]
         """
         Computes the truth value of ``self_i != other_i`` for each element of an array instance with the respective element of the array ``other``.
 
@@ -1078,7 +1078,7 @@ def __pos__(self: array, /) -> array:
         """
         ...
 
-    def __pow__(self: array, other: int | float | array, /) -> array:
+    def __pow__(self: array, other: int | float | Array, /) -> array:
         r"""
         Calculates an implementation-dependent approximation of exponentiation by raising each element (the base) of an array instance to the power of ``other_i`` (the exponent), where ``other_i`` is the corresponding element of the array ``other``.
 
@@ -1163,7 +1163,7 @@ def __setitem__(
         """
         ...
 
-    def __sub__(self: array, other: int | float | array, /) -> array:
+    def __sub__(self: array, other: int | float | Array, /) -> array:
         """
         Calculates the difference for each element of an array instance with the respective element of the array ``other``.
 
@@ -1192,7 +1192,7 @@ def __sub__(self: array, other: int | float | array, /) -> array:
         """
         ...
 
-    def __truediv__(self: array, other: int | float | array, /) -> array:
+    def __truediv__(self: array, other: int | float | Array, /) -> array:
         r"""
         Evaluates ``self_i / other_i`` for each element of an array instance with the respective element of the array ``other``.