ENH: Add annotations for generic.repeat and ndarray.repeat

Bas van Beek · Bas van Beek · commit 5f59e7729f0b · 2021-05-12T15:53:24.000+02:00
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
@@ -1563,12 +1563,6 @@ class _ArrayOrScalarCommon:
         keepdims: bool = ...,
     ) -> _NdArraySubClass: ...
 
-    def repeat(
-        self,
-        repeats: _ArrayLikeInt_co,
-        axis: Optional[SupportsIndex] = ...,
-    ) -> ndarray: ...
-
     @overload
     def round(
         self: _ArraySelf,
@@ -1829,6 +1823,12 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeType, _DType_co]):
         mode: _ModeKind = ...,
     ) -> _NdArraySubClass: ...
 
+    def repeat(
+        self,
+        repeats: _ArrayLikeInt_co,
+        axis: Optional[SupportsIndex] = ...,
+    ) -> ndarray[Any, _DType_co]: ...
+
     # Many of these special methods are irrelevant currently, since protocols
     # aren't supported yet. That said, I'm adding them for completeness.
     # https://docs.python.org/3/reference/datamodel.html
@@ -2821,6 +2821,12 @@ class generic(_ArrayOrScalarCommon):
         mode: _ModeKind = ...,
     ) -> _NdArraySubClass: ...
 
+    def repeat(
+        self: _ScalarType,
+        repeats: _ArrayLikeInt_co,
+        axis: Optional[SupportsIndex] = ...,
+    ) -> ndarray[Any, dtype[_ScalarType]]: ...
+
     def squeeze(
         self: _ScalarType, axis: Union[Literal[0], Tuple[()]] = ...
     ) -> _ScalarType: ...
