From e17a7095e2a7e8f76b3f40d9fff475e9b0a4a419 Mon Sep 17 00:00:00 2001 From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com> Date: Wed, 17 Apr 2024 11:59:17 +0800 Subject: [PATCH] Update base.py --- metagpt/rag/benchmark/base.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/metagpt/rag/benchmark/base.py b/metagpt/rag/benchmark/base.py index 1d8d9c600..c5e821330 100644 --- a/metagpt/rag/benchmark/base.py +++ b/metagpt/rag/benchmark/base.py @@ -160,16 +160,16 @@ class RAGBenchmark: @staticmethod def load_dataset(ds_names: list[str] = ["all"]): - infos = read_json_file(Path(EXAMPLE_BENCHMARK_PATH) / "dataset_info.json") + infos = read_json_file(EXAMPLE_BENCHMARK_PATH / "dataset_info.json") dataset_config = DatasetConfig( datasets=[ DatasetInfo( name=name, document_files=[ - Path(EXAMPLE_BENCHMARK_PATH) / name / file, + EXAMPLE_BENCHMARK_PATH / name / file for file in info["document_file"] ], - gt_info=read_json_file(Path(EXAMPLE_BENCHMARK_PATH) / name / info["gt_file"]), + gt_info=read_json_file(EXAMPLE_BENCHMARK_PATH / name / info["gt_file"]), ) for dataset_info in infos for name, info in dataset_info.items()