apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update rag_bm.py

Browse source
This commit is contained in:
YangQianli92 2024-04-16 14:48:31 +08:00 • committed by GitHub
parent 081c1b45c7
commit 52a062db60
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
examples/rag_bm.py
View file

@ -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: