[Fix] Fix pre-commit

This commit is contained in:
Mor-Li 2025-05-09 20:19:04 +08:00
parent d1da4a577c
commit 40c6c68162
2 changed files with 6 additions and 7 deletions

View File

@ -8,10 +8,9 @@ from datasets import Dataset
from huggingface_hub import hf_hub_download
from opencompass.datasets.base import BaseDataset
from opencompass.datasets.needlebench_v2.atc import (relationship_templates_en,
relationship_templates_zh_CN,
relationship_terms_en,
relationship_terms_zh_CN)
from opencompass.datasets.needlebench_v2.atc import (
relationship_templates_en, relationship_templates_zh_CN,
relationship_terms_en, relationship_terms_zh_CN)
from opencompass.registry import LOAD_DATASET

View File

@ -561,9 +561,9 @@ class NeedleBenchSummarizer(DefaultSummarizer):
class NeedleBenchSummarizerV2(NeedleBenchSummarizer):
"""NeedleBench summarizer V2 in OpenCompass.
This version calls save_results_to_plots with mean=True.
Args:
config (ConfigDict): The configuration object of the evaluation task. It's expected to be filled out at runtime.
dataset_abbrs (list[str], optional): Dataset abbreviations to be listed in the summary.
@ -572,7 +572,7 @@ class NeedleBenchSummarizerV2(NeedleBenchSummarizer):
'weights' if weighted average is needed.
prompt_db: A deprecated field.
"""
def summarize(
self,
output_path: str = None,