From dd6d427dc4eab567b2cc0daa0f0025515a834ffc Mon Sep 17 00:00:00 2001 From: Agus Date: Wed, 24 Jul 2024 14:17:09 +0200 Subject: [PATCH] Update src/distilabel/steps/generators/utils.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Gabriel Martín Blázquez --- src/distilabel/steps/generators/utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/distilabel/steps/generators/utils.py b/src/distilabel/steps/generators/utils.py index ef0177a018..969528c396 100644 --- a/src/distilabel/steps/generators/utils.py +++ b/src/distilabel/steps/generators/utils.py @@ -35,9 +35,10 @@ def make_generator_step( Args: dataset: The dataset to use in the `Pipeline`. batch_size: The batch_size, will default to the same used by the `GeneratorStep`s. - input_mappings: Applies the same as any other step. - output_mappings: Applies the same as any other step. - resources: Applies the same as any other step. + Defaults to `50`. + input_mappings: Applies the same as any other step. Defaults to `None`. + output_mappings: Applies the same as any other step. Defaults to `None`. + resources: Applies the same as any other step. Defaults to `StepResources()`. Raises: ValueError: If the format is different from the ones supported.