tensorflow fix tensorflow.data.Dataset's zip (#11357)

python · Feb 5, 2024 · 3600c80 · 3600c80
1 parent bf58d62
commit 3600c80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stubs/tensorflow/tensorflow/data/__init__.pyi b/stubs/tensorflow/tensorflow/data/__init__.pyi
@@ -226,7 +226,7 @@ class Dataset(ABC, Generic[_T1]):
     @overload
     @staticmethod
     def zip(
-        *args: Collection[Dataset[Any]] | ContainerGeneric[Dataset[Any]], name: str | None = None
+        *args: Dataset[Any] | Collection[Dataset[Any]] | ContainerGeneric[Dataset[Any]], name: str | None = None
     ) -> Dataset[tuple[Any, ...]]: ...
     @overload
     @staticmethod