Fix grpc test imports

pinecone-io · Nov 4, 2024 · 28dbadd · 28dbadd
1 parent 8f3adcd
commit 28dbadd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/integration/data/test_delete_future.py b/tests/integration/data/test_delete_future.py
@@ -1,9 +1,11 @@
 import os
 import pytest
 from pinecone import Vector
-from pinecone.grpc import GRPCDeleteResponse
 from ..helpers import poll_stats_for_namespace, random_string
 
+if os.environ.get("USE_GRPC") == "true":
+    from pinecone.grpc import GRPCDeleteResponse
+
 
 class TestDeleteFuture:
     @pytest.mark.skipif(

diff --git a/tests/integration/data/test_fetch_future.py b/tests/integration/data/test_fetch_future.py
@@ -1,6 +1,8 @@
 import os
 import pytest
-from pinecone.grpc import PineconeGrpcFuture
+
+if os.environ.get("USE_GRPC") == "true":
+    from pinecone.grpc import PineconeGrpcFuture
 
 
 @pytest.mark.skipif(