Skip to content

Commit

Permalink
Update src/distilabel/steps/generators/utils.py
Browse files Browse the repository at this point in the history
Co-authored-by: Gabriel Martín Blázquez <[email protected]>
  • Loading branch information
plaguss and gabrielmbmb authored Jul 24, 2024
1 parent 1acc0e1 commit dd6d427
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions src/distilabel/steps/generators/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit dd6d427

Please sign in to comment.