fix: argument name

sbintuitions · Dec 9, 2024 · 6428c19 · 6428c19
1 parent 672e8ba
commit 6428c19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/embedders/test_dp_sbert.py b/tests/embedders/test_dp_sbert.py
@@ -19,7 +19,7 @@ def test_encode(self):
         assert embeddings.shape == (OUTPUT_DIM,)
 
     def test_encode_with_prompt(self):
-        embeddings = self.model.encode("任意のテキスト", prompt="プロンプト")
+        embeddings = self.model.encode("任意のテキスト", prefix="プロンプト")
         assert isinstance(embeddings, np.ndarray)
         assert embeddings.shape == (OUTPUT_DIM,)