diff --git a/integrations/fastembed/src/haystack_integrations/components/embedders/fastembed/fastembed_text_SPLADE_embedder.py b/integrations/fastembed/src/haystack_integrations/components/embedders/fastembed/fastembed_text_SPLADE_embedder.py index 33700ac12..6703efa19 100644 --- a/integrations/fastembed/src/haystack_integrations/components/embedders/fastembed/fastembed_text_SPLADE_embedder.py +++ b/integrations/fastembed/src/haystack_integrations/components/embedders/fastembed/fastembed_text_SPLADE_embedder.py @@ -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.