Skip to content

Commit 8f3c996

Browse files
Fix Pinecone Cohere Integration System Test (#51466)
1 parent 94bacc9 commit 8f3c996

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def create_index():
5454
@task
5555
def transform_output(embedding_output) -> list[dict]:
5656
# Convert each embedding to a map with an ID and the embedding vector
57-
return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output.float_)]
57+
return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output)]
5858

5959
transformed_output = transform_output(embed_task.output)
6060

0 commit comments

Comments
 (0)