diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi index a81e060f6233..4972f3bf209c 100644 --- a/numpy/__init__.pyi +++ b/numpy/__init__.pyi @@ -4746,7 +4746,7 @@ class iinfo(Generic[_IntegerT_co]): class nditer: def __new__( cls, - op: ArrayLike | Sequence[ArrayLike], + op: ArrayLike | Sequence[ArrayLike | None], flags: None | Sequence[_NDIterFlagsKind] = ..., op_flags: None | Sequence[Sequence[_NDIterFlagsOp]] = ..., op_dtypes: DTypeLike | Sequence[DTypeLike] = ..., diff --git a/numpy/typing/tests/data/pass/nditer.py b/numpy/typing/tests/data/pass/nditer.py new file mode 100644 index 000000000000..25a5b44d7aec --- /dev/null +++ b/numpy/typing/tests/data/pass/nditer.py @@ -0,0 +1,4 @@ +import numpy as np + +arr = np.array([1]) +np.nditer([arr, None])