fix import by replacing GeneratorOutput with GeneratorStepOutput

argilla-io · Jan 10, 2025 · 16b89a0 · 16b89a0
1 parent e65894c
commit 16b89a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/docs/sections/how_to_guides/basic/task/generator_task.md b/docs/sections/how_to_guides/basic/task/generator_task.md
@@ -13,14 +13,13 @@ from typing_extensions import override
 
 from distilabel.steps.tasks.base import GeneratorTask
 from distilabel.steps.tasks.typing import ChatType
-from distilabel.steps.typing import GeneratorOutput
-
+from distilabel.steps.typing import GeneratorStepOutput
 
 class MyCustomTask(GeneratorTask):
     instruction: str
 
     @override
-    def process(self, offset: int = 0) -> GeneratorOutput:
+    def process(self, offset: int = 0) -> GeneratorStepOutput:
         output = self.llm.generate(
             inputs=[
                 [
@@ -79,11 +78,12 @@ from typing import Any, Dict, List, Union
 
 from distilabel.steps.tasks.base import GeneratorTask
 from distilabel.steps.tasks.typing import ChatType
+from distilabel.steps.typing import GeneratorStepOutput
 
 
 class MyCustomTask(GeneratorTask):
     @override
-    def process(self, offset: int = 0) -> GeneratorOutput:
+    def process(self, offset: int = 0) -> GeneratorStepOutput:
         output = self.llm.generate(
             inputs=[
                 [{"role": "user", "content": "Tell me a joke."}],

diff --git a/vllm b/vllm