diff --git a/src/datasets/arrow_dataset.py b/src/datasets/arrow_dataset.py
index 63871b29420..e887a0f025a 100644
--- a/src/datasets/arrow_dataset.py
+++ b/src/datasets/arrow_dataset.py
@@ -950,7 +950,7 @@ def from_generator(
         """Create a Dataset from a generator.
 
         Args:
-            generator (:obj:`Callable`): A callable object that returns an object that supports the `iter` protocol.
+            generator (:obj:`Callable`): A generator function that `yields` examples.
             features (:class:`Features`, optional): Dataset features.
             cache_dir (:obj:`str`, optional, default ``"~/.cache/huggingface/datasets"``): Directory to cache data.
             keep_in_memory (:obj:`bool`, default ``False``): Whether to copy the data in-memory.