Merge pull request #28913 from charris/backport-28908

charris · web-flow · commit 9e506595b1b4 · 2025-05-06T11:47:13.000-06:00
TYP: Fix non-existent ``CanIndex`` annotation in ``ndarray.setfield``
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
@@ -2551,7 +2551,7 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeT_co, _DType_co]):
     @overload  # (dtype: ?, type: type[T])
     def view(self, /, dtype: DTypeLike, type: type[_ArrayT]) -> _ArrayT: ...
 
-    def setfield(self, /, val: ArrayLike, dtype: DTypeLike, offset: CanIndex = 0) -> None: ...
+    def setfield(self, /, val: ArrayLike, dtype: DTypeLike, offset: SupportsIndex = 0) -> None: ...
     @overload
     def getfield(self, dtype: _DTypeLike[_SCT], offset: SupportsIndex = 0) -> NDArray[_SCT]: ...
     @overload
diff --git a/numpy/typing/tests/data/fail/array_like.pyi b/numpy/typing/tests/data/fail/array_like.pyi
@@ -14,3 +14,5 @@ scalar = np.int64(1)
 scalar.__array__(dtype=np.float64)  # E: No overload variant
 array = np.array([1])
 array.__array__(dtype=np.float64)  # E: No overload variant
+
+array.setfield(np.eye(1), np.int32, (0, 1))  # E: No overload variant
