Update tests/modules/text_embedding/test_embedding.py

Co-authored-by: Evaline Ju <[email protected]> Signed-off-by: Mynhardt Burger <[email protected]>
flaviabeo · Mar 8, 2024 · 4ef9dc1 · 4ef9dc1
1 parent 9cf0491
commit 4ef9dc1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/modules/text_embedding/test_embedding.py b/tests/modules/text_embedding/test_embedding.py
@@ -46,7 +46,7 @@
 ]
 
 QUERY = "What is foo bar?"
-QUERY_TOKEN_COUNT = 13 + 2 # [CLS] 13 normal [SEP]
+QUERY_TOKEN_COUNT = 13 + 2 # [CLS] Whatisfoobar? [SEP]
 
 QUERIES: List[str] = [
  "Who is foo?",