Skip to content

Commit

Permalink
Fix output component
Browse files Browse the repository at this point in the history
  • Loading branch information
Corentin authored Mar 13, 2024
1 parent 10ea129 commit 8e20cee
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ def warm_up(self):
model_name=self.model_name, cache_dir=self.cache_dir, threads=self.threads
)

@component.output_types(embedding=List[Dict[str, Union[List[int], List[float]]]])
@component.output_types(embedding=Dict[str, Union[List[int], List[float]]])
def run(self, text: str):
"""
Embeds text using the Fastembed model.
Expand Down

0 comments on commit 8e20cee

Please sign in to comment.