fix: Remove incorrect output types line (#8371)

* Remove incorrect output types line * Same update to zero shot text router * Update haystack/components/routers/zero_shot_text_router.py Co-authored-by: Amna Mubashar <[email protected]> --------- Co-authored-by: Amna Mubashar <[email protected]>
deepset-ai · Sep 17, 2024 · d76e654 · d76e654
1 parent 5071e47
commit d76e654
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/haystack/components/routers/transformers_text_router.py b/haystack/components/routers/transformers_text_router.py
@@ -179,8 +179,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "TransformersTextRouter":
             deserialize_hf_model_kwargs(data["init_parameters"]["huggingface_pipeline_kwargs"])
         return default_from_dict(cls, data)
 
-    @component.output_types(documents=Dict[str, str])
-    def run(self, text: str):
+    def run(self, text: str) -> Dict[str, str]:
         """
         Routes the text strings to different connections based on a category label.
 

diff --git a/haystack/components/routers/zero_shot_text_router.py b/haystack/components/routers/zero_shot_text_router.py
@@ -191,8 +191,7 @@ def from_dict(cls, data: Dict[str, Any]) -> "TransformersZeroShotTextRouter":
             deserialize_hf_model_kwargs(data["init_parameters"]["huggingface_pipeline_kwargs"])
         return default_from_dict(cls, data)
 
-    @component.output_types(documents=Dict[str, str])
-    def run(self, text: str):
+    def run(self, text: str) -> Dict[str, str]:
         """
         Routes the text strings to different connections based on a category label.