Skip to content

Commit 28dbadd

Browse files
committed
Fix grpc test imports
1 parent 8f3adcd commit 28dbadd

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

tests/integration/data/test_delete_future.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
import os
22
import pytest
33
from pinecone import Vector
4-
from pinecone.grpc import GRPCDeleteResponse
54
from ..helpers import poll_stats_for_namespace, random_string
65

6+
if os.environ.get("USE_GRPC") == "true":
7+
from pinecone.grpc import GRPCDeleteResponse
8+
79

810
class TestDeleteFuture:
911
@pytest.mark.skipif(

tests/integration/data/test_fetch_future.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import os
22
import pytest
3-
from pinecone.grpc import PineconeGrpcFuture
3+
4+
if os.environ.get("USE_GRPC") == "true":
5+
from pinecone.grpc import PineconeGrpcFuture
46

57

68
@pytest.mark.skipif(

0 commit comments

Comments
 (0)