remove rerank model must be required when retrieval_model is multiple (#6640)

This commit is contained in:
Jyong 2024-07-24 19:34:41 +08:00 committed by GitHub
parent b347a2f839
commit 600f13436d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -120,12 +120,6 @@ class DatasetConfigManager:
if not isinstance(config["dataset_configs"], dict):
raise ValueError("dataset_configs must be of object type")
if config["dataset_configs"]['retrieval_model'] == 'multiple':
if not config["dataset_configs"]['reranking_model']:
raise ValueError("reranking_model has not been set")
if not isinstance(config["dataset_configs"]['reranking_model'], dict):
raise ValueError("reranking_model must be of object type")
if not isinstance(config["dataset_configs"], dict):
raise ValueError("dataset_configs must be of object type")