Fix: retrieval test (#10662)
### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
parent
1d955507e9
commit
5fc59a3132
1 changed files with 2 additions and 1 deletions
|
|
@ -350,7 +350,8 @@ def retrieval_test():
|
||||||
float(req.get("similarity_threshold", 0.0)),
|
float(req.get("similarity_threshold", 0.0)),
|
||||||
float(req.get("vector_similarity_weight", 0.3)),
|
float(req.get("vector_similarity_weight", 0.3)),
|
||||||
top,
|
top,
|
||||||
doc_ids, rerank_mdl=rerank_mdl, highlight=req.get("highlight"),
|
doc_ids, rerank_mdl=rerank_mdl,
|
||||||
|
highlight=req.get("highlight", False),
|
||||||
rank_feature=labels
|
rank_feature=labels
|
||||||
)
|
)
|
||||||
if use_kg:
|
if use_kg:
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue