Skip to content

Commit

Permalink
Update add_raw_output description
Browse files Browse the repository at this point in the history
Co-authored-by: alvarobartt <[email protected]>
  • Loading branch information
gabrielmbmb and alvarobartt committed Jun 4, 2024
1 parent 578ccb2 commit 3a1be83
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion src/distilabel/steps/tasks/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,10 @@ class _Task(_Step, ABC):
group_generations: bool = False
add_raw_output: RuntimeParameter[bool] = Field(
default=True,
description="Whether to include the raw output of the LLM in the output.",
description=(
"Whether to include the raw output of the LLM in the key `raw_output_<TASK_NAME>`"
"of the `distilabel_metadata` dictionary output column"
),
)
num_generations: RuntimeParameter[int] = Field(
default=1, description="The number of generations to be produced per input."
Expand Down

0 comments on commit 3a1be83

Please sign in to comment.