TYP: Annotate untyped attributes and dunder methods (#31550)

jorenham · web-flow · commit 61371bf78fac · 2026-06-02T08:48:38.000-06:00
diff --git a/numpy/f2py/symbolic.pyi b/numpy/f2py/symbolic.pyi
@@ -1,6 +1,6 @@
 from collections.abc import Callable, Mapping
 from enum import Enum
-from typing import Any, Generic, Literal as L, Self, overload
+from typing import Any, Generic, Literal as L, Self, overload, override
 from typing_extensions import TypeVar
 
 __all__ = ["Expr"]
@@ -103,6 +103,10 @@ class Expr(Generic[_OpT_co, _DataT_co]):
     #
     def __init__(self, /, op: Op, data: _DataT_co) -> None: ...
 
+    #
+    @override
+    def __eq__(self, other: object, /) -> bool: ...
+
     #
     def __lt__(self, other: Expr, /) -> bool: ...
     def __le__(self, other: Expr, /) -> bool: ...
diff --git a/numpy/lib/_arrayterator_impl.pyi b/numpy/lib/_arrayterator_impl.pyi
@@ -34,6 +34,7 @@ class Arrayterator(np.ndarray[_ShapeT_co, _DTypeT_co]):
 
     #
     def __init__(self, /, var: np.ndarray[_ShapeT_co, _DTypeT_co], buf_size: int | None = None) -> None: ...
+    def __getattr__(self, attr: str, /) -> Any: ...
     def __getitem__(self, index: _AnyIndex, /) -> Arrayterator[_AnyShape, _DTypeT_co]: ...  # type: ignore[override]
     def __iter__(self) -> Generator[np.ndarray[_AnyShape, _DTypeT_co]]: ...  # pyrefly: ignore[bad-override]
 
diff --git a/numpy/lib/_index_tricks_impl.pyi b/numpy/lib/_index_tricks_impl.pyi
@@ -69,6 +69,8 @@ type _JustAnyShape = tuple[Never, Never, Never, Never, Never]  # workaround for
 ###
 
 class ndenumerate(Generic[_ScalarT_co]):
+    iter: np.flatiter[NDArray[_ScalarT_co]]
+
     @overload
     def __init__[ScalarT: np.generic](
         self: ndenumerate[ScalarT],
diff --git a/numpy/lib/_npyio_impl.pyi b/numpy/lib/_npyio_impl.pyi
@@ -64,8 +64,10 @@ class NpzFile(Mapping[str, NDArray[_ScalarT_co]]):
 
     zip: zipfile.ZipFile | None = None
     fid: IO[str] | None = None
+
     files: list[str]
     allow_pickle: bool
+    max_header_size: int
     pickle_kwargs: Mapping[str, Any] | None
     f: BagObj[NpzFile[_ScalarT_co]]
 
diff --git a/numpy/lib/_polynomial_impl.pyi b/numpy/lib/_polynomial_impl.pyi
@@ -9,6 +9,7 @@ from typing import (
     SupportsIndex,
     SupportsInt,
     overload,
+    override,
 )
 
 import numpy as np
@@ -143,6 +144,12 @@ class poly1d:
     def __truediv__(self, other: ArrayLike, /) -> Self: ...
     def __rtruediv__(self, other: ArrayLike, /) -> Self: ...
 
+    #
+    @override
+    def __eq__(self, other: poly1d, /) -> bool: ...  # type:ignore[override]
+    @override
+    def __ne__(self, other: poly1d, /) -> bool: ...  # type:ignore[override]
+
     #
     def deriv(self, /, m: ConvertibleToInt = 1) -> Self: ...
     def integ(self, /, m: ConvertibleToInt = 1, k: _ArrayLikeComplex_co | _ArrayLikeObject_co | None = 0) -> poly1d: ...
diff --git a/numpy/lib/_user_array_impl.pyi b/numpy/lib/_user_array_impl.pyi
@@ -223,3 +223,7 @@ class container(Generic[_ShapeT_co, _DTypeT_co]):
     def tobytes(self, /) -> bytes: ...
     def byteswap(self, /) -> Self: ...
     def astype[ScalarT: np.generic](self, /, typecode: _DTypeLike[ScalarT]) -> container[_ShapeT_co, np.dtype[ScalarT]]: ...
+
+    #
+    def __setattr__(self, attr: str, value: object, /) -> None: ...
+    def __getattr__(self, attr: str, /) -> Any: ...
diff --git a/numpy/ma/core.pyi b/numpy/ma/core.pyi
@@ -2780,6 +2780,12 @@ class MaskedArray(ndarray[_ShapeT_co, _DTypeT_co]):
 
     #
     @override
+    def __getstate__(self) -> tuple[Any, ...]: ...
+    @override
+    def __setstate__(self, state: tuple[Any, ...]) -> None: ...
+    @override
+    def __reduce__(self) -> tuple[Any, ...]: ...
+    @override
     def __deepcopy__(self, memo: dict[int, Any] | None = None) -> Self: ...
 
     # Keep `dtype` at the bottom to avoid name conflicts with `np.dtype`
diff --git a/numpy/ma/mrecords.pyi b/numpy/ma/mrecords.pyi
@@ -68,6 +68,10 @@ class MaskedRecords(MaskedArray[_ShapeT_co, _DTypeT_co], Generic[_ShapeT_co, _DT
     @override
     def __array_finalize__(self, obj: np.ndarray) -> None: ...  # type: ignore[override]  # pyright: ignore[reportIncompatibleMethodOverride]
     @override
+    def __getattribute__(self, attr: str, /) -> Any: ...
+    @override
+    def __setattr__(self, attr: str, val: Any, /) -> None: ...
+    @override
     def __getitem__(self, indx: str | _ToIndices, /) -> Incomplete: ...  # type: ignore[override]  # pyright: ignore[reportIncompatibleMethodOverride]
     @override
     def __setitem__(self, indx: str | _ToIndices, value: Incomplete, /) -> None: ...  # type: ignore[override]  # pyright: ignore[reportIncompatibleMethodOverride]

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ class Arrayterator(np.ndarray[_ShapeT_co, _DTypeT_co]):`
`34`	`34`
`35`	`35`	`#`
`36`	`36`	`def __init__(self, /, var: np.ndarray[_ShapeT_co, _DTypeT_co], buf_size: int \| None = None) -> None: ...`
	`37`	`+ def __getattr__(self, attr: str, /) -> Any: ...`
`37`	`38`	`def __getitem__(self, index: _AnyIndex, /) -> Arrayterator[_AnyShape, _DTypeT_co]: ... # type: ignore[override]`
`38`	`39`	`def __iter__(self) -> Generator[np.ndarray[_AnyShape, _DTypeT_co]]: ... # pyrefly: ignore[bad-override]`
`39`	`40`