Merge pull request #28044 from charris/backport-28039

charris · web-flow · commit 28a091a4efde · 2024-12-20T10:12:30.000-07:00
TYP: allow ``None`` in operand sequence of nditer
diff --git 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
@@ -0,0 +1,4 @@
+import numpy as np
+
+arr = np.array([1])
+np.nditer([arr, None])
