From 3600c805eb3059395ce2aa37ba32848b4465d817 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ho=C3=ABl=20Bagard?=
 <34478245+hoel-bagard@users.noreply.github.com>
Date: Mon, 5 Feb 2024 10:23:04 +0900
Subject: [PATCH] `tensorflow` fix `tensorflow.data.Dataset`'s zip (#11357)

---
 stubs/tensorflow/tensorflow/data/__init__.pyi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stubs/tensorflow/tensorflow/data/__init__.pyi b/stubs/tensorflow/tensorflow/data/__init__.pyi
index d1b551d95135..97f33b057ddc 100644
--- 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