TYP: allow None in operand sequence of nditer

Prevent type-hint errors when using `nditer` in an intended way (see https://numpy.org/doc/stable/reference/arrays.nditer.html#iterator-allocated-output-arrays). Fix #28038
numpy · Dec 20, 2024 · 554739e · 554739e
1 parent 31bc4c8
commit 554739e
Showing 1 changed file with 1 addition and 1 deletion.
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] = ...,