From 3a1be83df638a1a449d23b9c80d18e781c5805d8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gabriel=20Mart=C3=ADn=20Bl=C3=A1zquez?=
 <gmartinbdev@gmail.com>
Date: Tue, 4 Jun 2024 12:44:42 +0200
Subject: [PATCH] Update `add_raw_output` description

Co-authored-by: alvarobartt <alvarobartt@users.noreply.github.com>
---
 src/distilabel/steps/tasks/base.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/distilabel/steps/tasks/base.py b/src/distilabel/steps/tasks/base.py
index 39f0711dd5..df3b4243a4 100644
--- a/src/distilabel/steps/tasks/base.py
+++ b/src/distilabel/steps/tasks/base.py
@@ -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."