diff --git a/tests/embedders/test_dp_sbert.py b/tests/embedders/test_dp_sbert.py index 1d3e67b..e1cd4d2 100644 --- 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,)