From 5ade8d3cae2b83096169dab43faa46039eed2596 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mario=20=C5=A0a=C5=A1ko?= Date: Fri, 16 Sep 2022 13:26:56 +0200 Subject: [PATCH] Update src/datasets/arrow_dataset.py Co-authored-by: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com> --- src/datasets/arrow_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.