Add get_embedding_dimension to embedders (#6392)

GitOrigin-RevId: 3dc9b38c0ec25cb3c036d84d3131913839ed58ef
pathwaycom · May 7, 2024 · 05c8377 · 05c8377
1 parent 557393a
commit 05c8377
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pipelines/local/app.py b/examples/pipelines/local/app.py
@@ -63,7 +63,7 @@ def run(
     **kwargs,
 ):
     embedder = SentenceTransformerEmbedder(model=embedder_locator, device=device)
-    embedding_dimension = len(embedder.__wrapped__(""))
+    embedding_dimension = len(embedder.__wrapped__("."))
 
     documents = pw.io.jsonlines.read(
         data_dir,