From 52a062db6009fd9a13fcab0f2fc622e20362f204 Mon Sep 17 00:00:00 2001
From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com>
Date: Tue, 16 Apr 2024 14:48:31 +0800
Subject: [PATCH] Update rag_bm.py

---
 examples/rag_bm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/rag_bm.py b/examples/rag_bm.py
index a67eddc7c..bffe1e49b 100644
--- a/examples/rag_bm.py
+++ b/examples/rag_bm.py
@@ -45,7 +45,7 @@ class RAGExample:
         self.llm = get_rag_llm()
 
     async def rag_evaluate_pipeline(self, dataset_name: list[str] = ["CRUD"]):
-        dataset_config = self.benchmark.load_dataset()
+        dataset_config = self.benchmark.load_dataset(dataset_name)
 
         for dataset in dataset_config.datasets:
             if "all" in dataset_name or dataset.name in dataset_name: