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

remove type ignore

Browse source
This commit is contained in:
Xi Yan 2024-11-07 14:37:50 -08:00
parent f05db9a25c
commit 6da74262ef
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/apis/scoring_functions/scoring_functions.py
View file

@ -33,7 +33,7 @@ class ScoringConfigType(Enum):
@json_schema_type @json_schema_type
class LLMAsJudgeScoringFnParams(BaseModel): class LLMAsJudgeScoringFnParams(BaseModel):
type: Literal[ScoringConfigType.llm_as_judge.value] = ( # type: ignore type: Literal[ScoringConfigType.llm_as_judge.value] = (
ScoringConfigType.llm_as_judge.value ScoringConfigType.llm_as_judge.value
) )
judge_model: str judge_model: str
@ -46,7 +46,7 @@ class LLMAsJudgeScoringFnParams(BaseModel):
@json_schema_type @json_schema_type
class RegexParserScoringFnParams(BaseModel): class RegexParserScoringFnParams(BaseModel):
type: Literal[ScoringConfigType.regex_parser.value] = ( # type: ignore type: Literal[ScoringConfigType.regex_parser.value] = (
ScoringConfigType.regex_parser.value ScoringConfigType.regex_parser.value
) )
parsing_regexes: Optional[List[str]] = Field( parsing_regexes: Optional[List[str]] = Field(
@ -75,7 +75,7 @@ class ScoringFnDef(BaseModel):
return_type: ParamType = Field( return_type: ParamType = Field(
description="The return type of the deterministic function", description="The return type of the deterministic function",
) )
params: Optional[ScoringFnParams] = Field( # type: ignore params: Optional[ScoringFnParams] = Field(
description="The parameters for the scoring function for benchmark eval, these can be overridden for app eval", description="The parameters for the scoring function for benchmark eval, these can be overridden for app eval",
default=None, default=None,
) )