Spaces:
Build error
Build error
Sarp Bilgiç commited on
Commit ·
76c081e
1
Parent(s): 3525995
sparse tuple fix
Browse files
src/clients/sparse_embedding_client.py
CHANGED
|
@@ -11,9 +11,11 @@ class SparseEmbeddingClient:
|
|
| 11 |
)
|
| 12 |
logging.info("Sparse embedding model loaded successfully")
|
| 13 |
|
| 14 |
-
def embed_documents(self, documents: List[str]) ->
|
| 15 |
embeddings = list(self.model.embed(documents))
|
| 16 |
-
|
|
|
|
|
|
|
| 17 |
|
| 18 |
def embed_query(self, query: str) -> Tuple[List[int], List[float]]:
|
| 19 |
embedding = list(self.model.query_embed(query))[0]
|
|
|
|
| 11 |
)
|
| 12 |
logging.info("Sparse embedding model loaded successfully")
|
| 13 |
|
| 14 |
+
def embed_documents(self, documents: List[str]) -> Tuple[List[List[int]], List[List[float]]]:
|
| 15 |
embeddings = list(self.model.embed(documents))
|
| 16 |
+
all_indices = [emb.indices.tolist() for emb in embeddings]
|
| 17 |
+
all_values = [emb.values.tolist() for emb in embeddings]
|
| 18 |
+
return (all_indices, all_values)
|
| 19 |
|
| 20 |
def embed_query(self, query: str) -> Tuple[List[int], List[float]]:
|
| 21 |
embedding = list(self.model.query_embed(query))[0]
|