github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-04 04:07:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

pass list of strings to embed method in tf_hub (#3284)

Browse Source 
This fixes the below mentioned issue. Instead of simply passing the text
to `tensorflow_hub`, we convert it to a list and then pass it.
https://github.com/hwchase17/langchain/issues/3282

Co-authored-by: Prakhar Agarwal <i.prakhar-agarwal@devrev.ai>
This commit is contained in:
Prakhar Agarwal 2023-04-25 08:21:53 +05:30 committed by GitHub
parent 2b9f1cea4e
commit 53b14de636
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
langchain/embeddings/tensorflow_hub.py
View File

@ -66,5 +66,5 @@ class TensorflowHubEmbeddings(BaseModel, Embeddings):
Embeddings for the text.
"""
text = text.replace("\n", " ")
embedding = self.embed(text).numpy()[0]
embedding = self.embed([text]).numpy()[0]
return embedding.tolist()