🥅 np.polynomial workarounds for Any shape overload bugs

Author: jorenham

src/numpy-stubs/@test/static/accept/polynomial_series.pyi

@@ -79,11 +79,11 @@ assert_type(npp.polynomial.polyval3d(AR_f8, AR_f8, AR_f8, AR_i8), _Floating_nd)
 assert_type(npp.polynomial.polyval3d(AR_i8, AR_i8, AR_i8, AR_c16), _ComplexFloating_nd)
 assert_type(npp.polynomial.polyval3d(AR_O, AR_O, AR_O, AR_O), _Object_nd)
 
-assert_type(npp.polynomial.polyvalfromroots(AR_b, AR_b), _Floating_nd)
-assert_type(npp.polynomial.polyvalfromroots(AR_u4, AR_b), _Floating_nd)
-assert_type(npp.polynomial.polyvalfromroots(AR_i8, AR_i8), _Floating_nd)
-assert_type(npp.polynomial.polyvalfromroots(AR_f8, AR_i8), _Floating_nd)
-assert_type(npp.polynomial.polyvalfromroots(AR_i8, AR_c16), _ComplexFloating_nd)
+assert_type(npp.polynomial.polyvalfromroots(AR_b, AR_b), _Float64_nd)
+assert_type(npp.polynomial.polyvalfromroots(AR_u4, AR_b), _Float64_nd)
+assert_type(npp.polynomial.polyvalfromroots(AR_i8, AR_i8), _Float64_nd)
+assert_type(npp.polynomial.polyvalfromroots(AR_f8, AR_i8), _Float64_nd)
+assert_type(npp.polynomial.polyvalfromroots(AR_i8, AR_c16), _Complex128_nd)
 assert_type(npp.polynomial.polyvalfromroots(AR_O, AR_O), _Object_nd)
 
 assert_type(npp.polynomial.polyvander(AR_f8, 3), _Float64_nd)

src/numpy-stubs/polynomial/_polybase.pyi

@@ -58,16 +58,18 @@ class ABCPolyBase(abc.ABC):
     ) -> None: ...
 
     #
-    @overload
+    @overload  # workaround for microsoft/pyright#10232
     def __call__(self, /, arg: _PolyT) -> _PolyT: ...
     @overload
+    def __call__(self, /, arg: _nt._ToArray_nnd[_nt.co_complex]) -> _nt.Array[_nt.inexact64]: ...
+    @overload
     def __call__(self, /, arg: _nt.CoComplex_0d) -> _nt.inexact64: ...
     @overload
     def __call__(self, /, arg: _nt.CoComplex_1nd) -> _nt.Array[_nt.inexact64]: ...
     @overload
-    def __call__(self, /, arg: _nt.ToObject_0d) -> Any: ...
-    @overload
     def __call__(self, /, arg: _nt.ToObject_1nd) -> _nt.Array[np.object_]: ...
+    @overload
+    def __call__(self, /, arg: _nt.ToObject_nd) -> Any: ...
 
     #
     @override