phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-11 20:40:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pre-commit

Browse source
This commit is contained in:
Botao Chen 2025-03-10 20:01:22 -07:00
parent 90ed992138
commit 5319cef5af
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/scoring/basic/scoring.py
View file

@ -23,9 +23,9 @@ from llama_stack.providers.utils.common.data_schema_validator import (
from .config import BasicScoringConfig from .config import BasicScoringConfig
from .scoring_fn.equality_scoring_fn import EqualityScoringFn 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.regex_parser_scoring_fn import RegexParserScoringFn
from .scoring_fn.subset_of_scoring_fn import SubsetOfScoringFn 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] FIXED_FNS = [EqualityScoringFn, SubsetOfScoringFn, RegexParserScoringFn, RegexParserMathResponseScoringFn]

4
llama_stack/providers/inline/scoring/basic/utils/math_utils.py
View file

@ -11,14 +11,14 @@ from types import FrameType
from typing import Iterator, Optional, Sequence from typing import Iterator, Optional, Sequence
class TimeoutException(Exception): class TimeoutError(Exception):
pass pass
@contextlib.contextmanager @contextlib.contextmanager
def time_limit(seconds: float) -> Iterator[None]: def time_limit(seconds: float) -> Iterator[None]:
def signal_handler(signum: int, frame: Optional[FrameType]) -> 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.setitimer(signal.ITIMER_REAL, seconds)
signal.signal(signal.SIGALRM, signal_handler) signal.signal(signal.SIGALRM, signal_handler)