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

naming fix

Browse source
This commit is contained in:
Xi Yan 2024-11-12 10:29:25 -05:00
parent a15a4a4be2
commit ba1d50c67b
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -34,8 +34,8 @@ class BasicScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
self.scoring_fn_id_impls = {} self.scoring_fn_id_impls = {}
async def initialize(self) -> None: async def initialize(self) -> None:
for x in FIXED_FNS: for fn in FIXED_FNS:
impl = x() impl = fn()
for fn_defs in impl.get_supported_scoring_fn_defs(): for fn_defs in impl.get_supported_scoring_fn_defs():
self.scoring_fn_id_impls[fn_defs.identifier] = impl self.scoring_fn_id_impls[fn_defs.identifier] = impl

6
llama_stack/providers/inline/scoring/llm_as_judge/scoring.py
View file

@ -40,8 +40,8 @@ class LlmAsJudgeScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
self.scoring_fn_id_impls = {} self.scoring_fn_id_impls = {}
async def initialize(self) -> None: async def initialize(self) -> None:
for x in LLM_JUDGE_FNS: for fn in LLM_JUDGE_FNS:
impl = x(inference_api=self.inference_api) impl = fn(inference_api=self.inference_api)
for fn_defs in impl.get_supported_scoring_fn_defs(): for fn_defs in impl.get_supported_scoring_fn_defs():
self.scoring_fn_id_impls[fn_defs.identifier] = impl self.scoring_fn_id_impls[fn_defs.identifier] = impl
self.llm_as_judge_fn = impl self.llm_as_judge_fn = impl
@ -58,7 +58,7 @@ class LlmAsJudgeScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
for f in scoring_fn_defs_list: for f in scoring_fn_defs_list:
assert f.identifier.startswith( assert f.identifier.startswith(
"llm-as-judge" "llm-as-judge"
), "All meta-reference scoring fn must have identifier prefixed with 'meta-reference'! " ), "All llm-as-judge scoring fn must have identifier prefixed with 'llm-as-judge'! "
return scoring_fn_defs_list return scoring_fn_defs_list