Skip to content

Commit

Permalink
Black formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
jhamon committed Jul 19, 2024
1 parent 894b671 commit 5af0e15
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions tests/integration/inference/conftest.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import pytest
from ..helpers import get_environment_var


@pytest.fixture()
def api_key():
return get_environment_var("PINECONE_API_KEY")
3 changes: 2 additions & 1 deletion tests/integration/inference/test_embed.py
Original file line number Diff line number Diff line change
@@ -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)
Expand Down Expand Up @@ -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
assert embeddings.get("model") == embedding_model

0 comments on commit 5af0e15

Please sign in to comment.