From 28bc0f53356ab5d2c95569f2a34033039048ef3e Mon Sep 17 00:00:00 2001 From: Mallori Harrell Date: Tue, 3 Oct 2023 12:00:37 -0500 Subject: [PATCH] add missing return --- unstructured/embed/huggingface.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/unstructured/embed/huggingface.py b/unstructured/embed/huggingface.py index 2432d1877e..e43b617b3f 100644 --- a/unstructured/embed/huggingface.py +++ b/unstructured/embed/huggingface.py @@ -8,8 +8,6 @@ Element, ) from unstructured.embed.interfaces import BaseEmbeddingEncoder -from unstructured.ingest.error import EmbeddingEncoderConnectionError -from unstructured.utils import requires_dependencies class HuggingFaceEmbeddingEncoder(BaseEmbeddingEncoder): @@ -70,3 +68,4 @@ def new_to_dict(self): element.embeddings = embeddings[i] elements_w_embedding.append(element) element.to_dict = types.MethodType(new_to_dict, element) + return elements