From 6f02360b79fc0bc7cc7983a36bb983f01392d7ea Mon Sep 17 00:00:00 2001 From: vatsrahul1001 Date: Fri, 6 Jun 2025 12:11:07 +0530 Subject: [PATCH] fix pinecone system test --- .../pinecone/tests/system/pinecone/example_pinecone_cohere.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py index 496c0a40154bf..2e74fe870a07d 100644 --- a/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py +++ b/providers/pinecone/tests/system/pinecone/example_pinecone_cohere.py @@ -54,7 +54,7 @@ def create_index(): @task def transform_output(embedding_output) -> list[dict]: # Convert each embedding to a map with an ID and the embedding vector - return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output.float_)] + return [dict(id=str(i), values=embedding) for i, embedding in enumerate(embedding_output)] transformed_output = transform_output(embed_task.output)