fix: missed rerank_mode when convert to DatasetEntity (#7269)

This commit is contained in:
pp 2024-08-15 00:41:12 +08:00 committed by GitHub
parent ba79088ffc
commit 101db126c8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -93,6 +93,7 @@ class DatasetConfigManager:
reranking_model=dataset_configs.get('reranking_model'),
weights=dataset_configs.get('weights'),
reranking_enabled=dataset_configs.get('reranking_enabled', True),
rerank_mode=dataset_configs["reranking_mode"],
)
)