Skip to content

Commit

Permalink
add missing return
Browse files Browse the repository at this point in the history
  • Loading branch information
Mallori Harrell authored and Mallori Harrell committed Oct 3, 2023
1 parent 8bb69a0 commit 28bc0f5
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions unstructured/embed/huggingface.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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

0 comments on commit 28bc0f5

Please sign in to comment.