summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBas van Beek <b.f.van.beek@vu.nl>2021-10-01 10:29:00 +0200
committerBas van Beek <b.f.van.beek@vu.nl>2021-10-01 10:29:00 +0200
commitf5a8e63f2131464664d9d541f4f2bb4b0521bacf (patch)
treed5576316f0e3d50d094f3d726f2b752eac86e233
parent6e2cfa4d1482d2153131319f81d1db84753d8a1c (diff)
downloadnumpy-f5a8e63f2131464664d9d541f4f2bb4b0521bacf.tar.gz
MAINT: Remove `slice` from the accepted `np.place` mask-types
`np.place` uses the `np.core.multiarray._insert` function under the hood, not to be confused with `np.insert` (which does accept slices)
-rw-r--r--numpy/lib/function_base.pyi2
-rw-r--r--numpy/typing/tests/data/fail/lib_function_base.pyi1
2 files changed, 2 insertions, 1 deletions
diff --git a/numpy/lib/function_base.pyi b/numpy/lib/function_base.pyi
index 71fb189f4..cbbc87e65 100644
--- a/numpy/lib/function_base.pyi
+++ b/numpy/lib/function_base.pyi
@@ -300,7 +300,7 @@ def extract(condition: ArrayLike, arr: _ArrayLike[_SCT]) -> NDArray[_SCT]: ...
@overload
def extract(condition: ArrayLike, arr: ArrayLike) -> NDArray[Any]: ...
-def place(arr: NDArray[Any], mask: slice | ArrayLike, vals: Any) -> None: ...
+def place(arr: NDArray[Any], mask: ArrayLike, vals: Any) -> None: ...
def disp(
mesg: object,
diff --git a/numpy/typing/tests/data/fail/lib_function_base.pyi b/numpy/typing/tests/data/fail/lib_function_base.pyi
index faea2d981..019bd7f01 100644
--- a/numpy/typing/tests/data/fail/lib_function_base.pyi
+++ b/numpy/typing/tests/data/fail/lib_function_base.pyi
@@ -16,3 +16,4 @@ np.trim_zeros(1) # E: incompatible type
np.place(1, [True], 1.5) # E: incompatible type
np.vectorize(1) # E: incompatible type
np.add_newdoc("__main__", 1.5, "docstring") # E: incompatible type
+np.place(AR_f8, slice(None), 5) # E: incompatible type