summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--numpy/__init__.pyi1
-rw-r--r--numpy/typing/tests/data/fail/ndarray_misc.pyi2
-rw-r--r--numpy/typing/tests/data/reveal/ndarray_misc.pyi4
3 files changed, 6 insertions, 1 deletions
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
index 29fbc79b7..8d82013e2 100644
--- a/numpy/__init__.pyi
+++ b/numpy/__init__.pyi
@@ -1524,7 +1524,6 @@ class ndarray(_ArrayOrScalarCommon, Generic[_ShapeType, _DType_co]):
kwargs: Mapping[str, Any],
) -> Any: ...
- @property
def __array_finalize__(self, obj: None | NDArray[Any], /) -> None: ...
def __array_wrap__(
diff --git a/numpy/typing/tests/data/fail/ndarray_misc.pyi b/numpy/typing/tests/data/fail/ndarray_misc.pyi
index 8320a44f3..77bd9a44e 100644
--- a/numpy/typing/tests/data/fail/ndarray_misc.pyi
+++ b/numpy/typing/tests/data/fail/ndarray_misc.pyi
@@ -39,3 +39,5 @@ AR_b.__index__() # E: Invalid self argument
AR_f8[1.5] # E: No overload variant
AR_f8["field_a"] # E: No overload variant
AR_f8[["field_a", "field_b"]] # E: Invalid index type
+
+AR_f8.__array_finalize__(object()) # E: incompatible type
diff --git a/numpy/typing/tests/data/reveal/ndarray_misc.pyi b/numpy/typing/tests/data/reveal/ndarray_misc.pyi
index f91d6351b..c9a42b3e8 100644
--- a/numpy/typing/tests/data/reveal/ndarray_misc.pyi
+++ b/numpy/typing/tests/data/reveal/ndarray_misc.pyi
@@ -212,3 +212,7 @@ reveal_type(AR_f8.dump("test_file")) # E: None
reveal_type(AR_f8.dump(b"test_file")) # E: None
with open("test_file", "wb") as f:
reveal_type(AR_f8.dump(f)) # E: None
+
+reveal_type(AR_f8.__array_finalize__(None)) # E: None
+reveal_type(AR_f8.__array_finalize__(B)) # E: None
+reveal_type(AR_f8.__array_finalize__(AR_f8)) # E: None