[tf.data] More actionable error message when passing a list to `Dataset.zip()`.

PiperOrigin-RevId: 170716623

1 files changed, 10 insertions, 0 deletions

diff --git a/tensorflow/python/data/ops/dataset_ops.py b/tensorflow/python/data/ops/dataset_ops.py
index 9bcc83e8c5..aaea0f5db0 100644
--- a/tensorflow/python/data/ops/dataset_ops.py
+++ b/tensorflow/python/data/ops/dataset_ops.py
@@ -933,6 +933,16 @@ class ZipDataset(Dataset):
   def __init__(self, datasets):
     """See `Dataset.zip()` for details."""
     super(ZipDataset, self).__init__()
+    for ds in nest.flatten(datasets):
+      if not isinstance(ds, Dataset):
+        if isinstance(ds, list):
+          message = ("The argument to `Dataset.zip()` must be a nested "
+                     "structure of `Dataset` objects. Nested structures do not "
+                     "support Python lists; please use a tuple instead.")
+        else:
+          message = ("The argument to `Dataset.zip()` must be a nested "
+                     "structure of `Dataset` objects.")
+        raise TypeError(message)
     self._datasets = datasets
 
   def _as_variant_tensor(self):