From 0af7b2c3069122e6dc1a64cae98d725eddeb05be Mon Sep 17 00:00:00 2001
From: mariosasko <mariosasko777@gmail.com>
Date: Thu, 8 Sep 2022 17:12:00 +0200
Subject: [PATCH] Add Returns to docstring

---
 src/datasets/arrow_dataset.py | 3 +++
 1 file changed, 3 insertions(+)

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