internal.

PiperOrigin-RevId: 886200959

Author: iindyk

grain/_src/python/dataset/dataset.py

@@ -1708,6 +1708,7 @@ def __init__(self, parent: IterDataset[T], options: base.DatasetOptions):
     self.options = options
 
   def __iter__(self) -> DatasetIterator[T]:
+    """Returns an iterator for the dataset."""
     result = self._parent.__iter__()
     # The parent iterator options are merged from the entire subtree. Merge
     # them with the latest options and update the subtree options.
@@ -1737,7 +1738,7 @@ class _OutputIterDataset(IterDataset[T]):
   """Dataset that is injected at the end of every pipeline."""
 
   def __iter__(self) -> DatasetIterator[T]:
-    """Performs any injection that needs to happen at the end of the pipeline."""
+    """Performs injections that need to happen at the end of the pipeline."""
 
     # Loaded lazily due to a circular dependency (dataset <-> prefetch).
     # pylint: disable=g-import-not-at-top