Fix tests

argilla-io · Jun 4, 2024 · aac845c · aac845c
1 parent 74064ff
commit aac845c
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/src/distilabel/steps/tasks/base.py b/src/distilabel/steps/tasks/base.py
@@ -56,7 +56,7 @@ class _Task(_Step, ABC):
         default=True,
         description=(
             "Whether to include the raw output of the LLM in the key `raw_output_<TASK_NAME>`"
-            "of the `distilabel_metadata` dictionary output column"
+            " of the `distilabel_metadata` dictionary output column"
         ),
     )
     num_generations: RuntimeParameter[int] = Field(

diff --git a/tests/unit/steps/tasks/evol_instruct/test_base.py b/tests/unit/steps/tasks/evol_instruct/test_base.py
@@ -164,7 +164,7 @@ def test_serialization(self, dummy_llm: LLM) -> None:
                     ],
                 },
                 {
-                    "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",
                     "name": "add_raw_output",
                     "optional": True,
                 },

diff --git a/tests/unit/steps/tasks/evol_instruct/test_generator.py b/tests/unit/steps/tasks/evol_instruct/test_generator.py
@@ -159,7 +159,7 @@ def test_serialization(self, dummy_llm: LLM) -> None:
                     ],
                 },
                 {
-                    "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",
                     "name": "add_raw_output",
                     "optional": True,
                 },

diff --git a/tests/unit/steps/tasks/evol_quality/test_base.py b/tests/unit/steps/tasks/evol_quality/test_base.py
@@ -116,7 +116,7 @@ def test_serialization(self, dummy_llm: LLM) -> None:
                     ],
                 },
                 {
-                    "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",
                     "name": "add_raw_output",
                     "optional": True,
                 },

diff --git a/tests/unit/steps/tasks/test_base.py b/tests/unit/steps/tasks/test_base.py
@@ -233,7 +233,7 @@ def test_serialization(self) -> None:
                     ],
                 },
                 {
-                    "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",
                     "name": "add_raw_output",
                     "optional": True,
                 },