xiaoxue_admin/zvec
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: enlarge topK limit (#348)

Browse Source
This commit is contained in:
Qinren Zhou
2026-04-17 17:51:07 +08:00
committed by GitHub
parent 3f9ebbbe96
commit f602ed30ce
7 changed files with 43 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/tests/test_embedding.py
+4 -4
View File
@@ -1168,8 +1168,8 @@ class TestDefaultLocalDenseEmbedding:
return_value="/path/to/model",
):
mock_ms = Mock()
mock_require_module.side_effect = (
lambda m: mock_st if m == "sentence_transformers" else mock_ms
mock_require_module.side_effect = lambda m: (
mock_st if m == "sentence_transformers" else mock_ms
)
emb_func_ms = DefaultLocalDenseEmbedding(model_source="modelscope")
assert (
@@ -1635,8 +1635,8 @@ class TestDefaultLocalSparseEmbedding:
"modelscope.hub.snapshot_download.snapshot_download",
return_value="/cache/splade-cocondenser",
):
mock_require_module.side_effect = (
lambda m: mock_st if m == "sentence_transformers" else mock_ms
mock_require_module.side_effect = lambda m: (
mock_st if m == "sentence_transformers" else mock_ms
)
sparse_emb = DefaultLocalSparseEmbedding(model_source="modelscope")