diff --git a/src/datasets/arrow_dataset.py b/src/datasets/arrow_dataset.py
index d62be679656..3d2b28202a6 100644
--- a/src/datasets/arrow_dataset.py
+++ b/src/datasets/arrow_dataset.py
@@ -954,6 +954,9 @@ def from_generator(
             features (:class:`Features`, optional): Dataset features.
             gen_kwargs(:obj:`dict`, optional): Keyword arguments to be passed to the `generator` callable.
 
+        Returns:
+            :class:`Dataset`
+
         Example:
 
         ```py