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

This commit is contained in:
Hoël Bagard
2024-02-05 10:23:04 +09:00
committed by GitHub
parent bf58d62e74
commit 3600c805eb

View File

@@ -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