From 6cd454a827d59cbc07d0cfdbf2d7f60ebb250042 Mon Sep 17 00:00:00 2001 From: David Berenstein Date: Wed, 12 Jun 2024 09:18:54 +0200 Subject: [PATCH] Update docs/sections/how_to_guides/basic/step/index.md Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com> --- docs/sections/how_to_guides/basic/step/index.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/sections/how_to_guides/basic/step/index.md b/docs/sections/how_to_guides/basic/step/index.md index 42b169a67e..b418a4f292 100644 --- a/docs/sections/how_to_guides/basic/step/index.md +++ b/docs/sections/how_to_guides/basic/step/index.md @@ -123,7 +123,7 @@ We can define a custom step by creating a new subclass of the [`Step`][distilabe from distilabel.steps.typing import StepOutput @step(inputs=[...], outputs=[...]) - def CustomStep(inputs: StepInput) - StepOutput: + def CustomStep(inputs: StepInput) -> StepOutput: for input in inputs: ... yield inputs