.

pinecone-io · Sep 13, 2024 · 819586d · 819586d
1 parent d117b6f
commit 819586d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration/inference/test_rerank.py b/tests/integration/inference/test_rerank.py
@@ -3,7 +3,7 @@
 
 
 class TestInferencePluginRerank:
-    def test_rerank(api_key):
+    def test_rerank(self, api_key):
         pc = Pinecone(api_key=api_key)
 
         model = "bge-reranker-v2-m3"
@@ -21,7 +21,7 @@ def test_rerank(api_key):
         assert isinstance(result.usage.rerank_units, int)
         assert result.usage.rerank_units == 1
 
-    def test_rerank_grpc(api_key):
+    def test_rerank_grpc(self, api_key):
         pc = PineconeGRPC(api_key=api_key)
 
         model = "bge-reranker-v2-m3"