From dcbf899369dae31af8f1ecb9a59eb2c50cf2aad4 Mon Sep 17 00:00:00 2001 From: Linchen Xiao Date: Wed, 23 Apr 2025 11:10:30 +0800 Subject: [PATCH] [Bug] Fix SmolInsturct logger import (#2036) --- opencompass/datasets/smolinstruct.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/opencompass/datasets/smolinstruct.py b/opencompass/datasets/smolinstruct.py index cd3b21ed..e9577335 100644 --- a/opencompass/datasets/smolinstruct.py +++ b/opencompass/datasets/smolinstruct.py @@ -9,6 +9,7 @@ from nltk.translate.meteor_score import meteor_score from opencompass.openicl.icl_evaluator.icl_base_evaluator import BaseEvaluator from opencompass.registry import (ICL_EVALUATORS, LOAD_DATASET, TEXT_POSTPROCESSORS) +from opencompass.utils import get_logger from .base import BaseDataset @@ -413,8 +414,9 @@ class MeteorEvaluator(BaseEvaluator): score = (meteor_score([ans.split()], pred.split()) if ans and pred else 0.0) except AttributeError: - logging.error(f'Failed to compute METEOR' - f"score:\npred='{pred}'\nans='{ans}'") + self.logger = get_logger() + self.logger.warning(f'Failed to compute METEOR' + f"score:\npred='{pred}'\nans='{ans}'") score = 0.0 avg_score += score detail = {'pred': pred, 'answer': ans, 'score': score}