Fix DummyTextEmbedder

sbintuitions · Jun 17, 2024 · 32fed50 · 32fed50
1 parent 5458301
commit 32fed50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -25,7 +25,7 @@ def pytest_collection_modifyitems(config: pytest.Config, items: pytest.Parser):
 
 
 class DummyTextEmbedder(TextEmbedder):
-    def encode(self, text: str | list[str]) -> np.ndarray:
+    def encode(self, text: str | list[str], prompt: str | None = None) -> np.ndarray:
         if isinstance(text, str):
             batch_size = 1
         else: