Adding default value to SentenceTransformer wrapper

GitOrigin-RevId: 92241e5ded4f140c8ec4ffab3a3c0e60ff48204c
pathwaycom · Feb 1, 2024 · 13d97de · 13d97de
1 parent 1854732
commit 13d97de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/pathway/xpacks/llm/embedders.py b/python/pathway/xpacks/llm/embedders.py
@@ -197,7 +197,7 @@ class SentenceTransformerEmbedder(pw.UDFSync):
     def __init__(
         self,
         model: str,
-        call_kwargs: dict,
+        call_kwargs: dict = {},
         device: str = "cpu",
         **sentencetransformer_kwargs,
     ):