Update rag_bm.py

This commit is contained in:
YangQianli92 2024-04-17 11:06:40 +08:00 committed by GitHub
parent ac31e67607
commit 9350fd9c37
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -47,7 +47,7 @@ class RAGExample:
self.embedding = get_rag_embedding()
self.llm = get_rag_llm()
async def rag_evaluate_pipeline(self, dataset_name: list[str] = ["RGB_en"]):
async def rag_evaluate_pipeline(self, dataset_name: list[str] = ["all"]):
dataset_config = self.benchmark.load_dataset(dataset_name)
for dataset in dataset_config.datasets: