ObjectArray: use complex for scalars

Author: inducer

pytools/obj_array.py

@@ -203,55 +203,55 @@ def __add__(self, other: Self, /) -> Self: ...
         @overload
         def __add__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __add__(self, other: float, /) -> Self: ...
+        def __add__(self, other: complex, /) -> Self: ...
         @overload
         def __radd__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __radd__(self, other: float, /) -> Self: ...
+        def __radd__(self, other: complex, /) -> Self: ...
 
         @overload
         def __sub__(self, other: Self, /) -> Self: ...
         @overload
         def __sub__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __sub__(self, other: float, /) -> Self: ...
+        def __sub__(self, other: complex, /) -> Self: ...
         @overload
         def __rsub__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __rsub__(self, other: float, /) -> Self: ...
+        def __rsub__(self, other: complex, /) -> Self: ...
 
         @overload
         def __mul__(self, other: Self, /) -> Self: ...
         @overload
         def __mul__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __mul__(self, other: float, /) -> Self: ...
+        def __mul__(self, other: complex, /) -> Self: ...
         @overload
         def __rmul__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __rmul__(self, other: float, /) -> Self: ...
+        def __rmul__(self, other: complex, /) -> Self: ...
 
         @overload
         def __truediv__(self, other: Self, /) -> Self: ...
         @overload
         def __truediv__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __truediv__(self, other: float, /) -> Self: ...
+        def __truediv__(self, other: complex, /) -> Self: ...
         @overload
         def __rtruediv__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __rtruediv__(self, other: float, /) -> Self: ...
+        def __rtruediv__(self, other: complex, /) -> Self: ...
 
         @overload
         def __pow__(self, other: Self, /) -> Self: ...
         @overload
         def __pow__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __pow__(self, other: float, /) -> Self: ...
+        def __pow__(self, other: complex, /) -> Self: ...
         @overload
         def __rpow__(self, other: T, /) -> Self: ...  # pyright: ignore[reportGeneralTypeIssues]
         @overload
-        def __rpow__(self, other: float, /) -> Self: ...
+        def __rpow__(self, other: complex, /) -> Self: ...
 
         @overload
         def __matmul__(