Skip to content

Commit

Permalink
Test fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jhamon committed Nov 4, 2024
1 parent b26f05e commit 8f3adcd
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
2 changes: 1 addition & 1 deletion pinecone/grpc/future.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ def _sync_state(self, grpc_future):

def set_result(self, result):
if self._result_transformer:
result = self.result_transformer(result)
result = self._result_transformer(result)
return super().set_result(result)

def cancel(self):
Expand Down
6 changes: 4 additions & 2 deletions tests/integration/data/test_delete_future.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,16 @@
import pytest
from pinecone import Vector
from pinecone.grpc import GRPCDeleteResponse
from ..helpers import poll_stats_for_namespace
from ..helpers import poll_stats_for_namespace, random_string


class TestDeleteFuture:
@pytest.mark.skipif(
os.getenv("USE_GRPC") != "true", reason="PineconeGrpcFutures only returned from grpc client"
)
def test_delete_future(self, idx, namespace):
def test_delete_future(self, idx):
namespace = random_string(10)

idx.upsert(
vectors=[
Vector(id="id1", values=[0.1, 0.2]),
Expand Down

0 comments on commit 8f3adcd

Please sign in to comment.