Skip to content

Commit

Permalink
Update haystack/components/routers/zero_shot_text_router.py
Browse files Browse the repository at this point in the history
Co-authored-by: Amna Mubashar <[email protected]>
  • Loading branch information
sjrl and Amnah199 authored Sep 16, 2024
1 parent 13645b0 commit 676d2ac
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion haystack/components/routers/zero_shot_text_router.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +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)

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.
Expand Down

0 comments on commit 676d2ac

Please sign in to comment.