From 5319cef5af53f717764d0eb4fb7385765eeb4cc3 Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Mon, 10 Mar 2025 20:01:22 -0700 Subject: [PATCH] fix pre-commit --- llama_stack/providers/inline/scoring/basic/scoring.py | 2 +- .../providers/inline/scoring/basic/utils/math_utils.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/inline/scoring/basic/scoring.py b/llama_stack/providers/inline/scoring/basic/scoring.py index cbaf015d6..00945b99d 100644 --- a/llama_stack/providers/inline/scoring/basic/scoring.py +++ b/llama_stack/providers/inline/scoring/basic/scoring.py @@ -23,9 +23,9 @@ from llama_stack.providers.utils.common.data_schema_validator import ( from .config import BasicScoringConfig from .scoring_fn.equality_scoring_fn import EqualityScoringFn +from .scoring_fn.regex_parser_math_response_scoring_fn import RegexParserMathResponseScoringFn from .scoring_fn.regex_parser_scoring_fn import RegexParserScoringFn from .scoring_fn.subset_of_scoring_fn import SubsetOfScoringFn -from .scoring_fn.regex_parser_math_response_scoring_fn import RegexParserMathResponseScoringFn FIXED_FNS = [EqualityScoringFn, SubsetOfScoringFn, RegexParserScoringFn, RegexParserMathResponseScoringFn] diff --git a/llama_stack/providers/inline/scoring/basic/utils/math_utils.py b/llama_stack/providers/inline/scoring/basic/utils/math_utils.py index f108886a2..2fe5650c9 100644 --- a/llama_stack/providers/inline/scoring/basic/utils/math_utils.py +++ b/llama_stack/providers/inline/scoring/basic/utils/math_utils.py @@ -11,14 +11,14 @@ from types import FrameType from typing import Iterator, Optional, Sequence -class TimeoutException(Exception): +class TimeoutError(Exception): pass @contextlib.contextmanager def time_limit(seconds: float) -> Iterator[None]: def signal_handler(signum: int, frame: Optional[FrameType]) -> None: - raise TimeoutException("Timed out!") + raise TimeoutError("Timed out!") signal.setitimer(signal.ITIMER_REAL, seconds) signal.signal(signal.SIGALRM, signal_handler)