From 5af0e15cc5a3913d8daaf5975179beb008d83885 Mon Sep 17 00:00:00 2001 From: Jen Hamon Date: Fri, 19 Jul 2024 11:10:26 -0400 Subject: [PATCH] Black formatting --- tests/integration/inference/conftest.py | 1 + tests/integration/inference/test_embed.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/integration/inference/conftest.py b/tests/integration/inference/conftest.py index 9d967dd3..2c345033 100644 --- a/tests/integration/inference/conftest.py +++ b/tests/integration/inference/conftest.py @@ -1,6 +1,7 @@ import pytest from ..helpers import get_environment_var + @pytest.fixture() def api_key(): return get_environment_var("PINECONE_API_KEY") diff --git a/tests/integration/inference/test_embed.py b/tests/integration/inference/test_embed.py index 4b599fb2..18e30c0c 100644 --- a/tests/integration/inference/test_embed.py +++ b/tests/integration/inference/test_embed.py @@ -1,6 +1,7 @@ from pinecone import Pinecone from pinecone.grpc import PineconeGRPC + class TestInferencePlugin: def test_embed(self, api_key): pc = Pinecone(api_key=api_key) @@ -30,4 +31,4 @@ def test_embed_grpc(self, api_key): assert len(embeddings.get("data")) == 2 assert len(embeddings.get("data")[0]["values"]) == 1024 assert len(embeddings.get("data")[1]["values"]) == 1024 - assert embeddings.get("model") == embedding_model \ No newline at end of file + assert embeddings.get("model") == embedding_model