From 0d8de1c76841fd28591918dd82ef70948cccf47b Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Sun, 17 Nov 2024 21:16:10 -0800 Subject: [PATCH] condition --- llama_stack/providers/inline/eval/meta_reference/eval.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/eval/meta_reference/eval.py b/llama_stack/providers/inline/eval/meta_reference/eval.py index 3e9d22546..7c3c27c77 100644 --- a/llama_stack/providers/inline/eval/meta_reference/eval.py +++ b/llama_stack/providers/inline/eval/meta_reference/eval.py @@ -226,9 +226,10 @@ class MetaReferenceEvalImpl(Eval, EvalTasksProtocolPrivate): candidate = task_config.eval_candidate if candidate.type == "agent": generations = await self._run_agent_generation(input_rows, task_config) - - if candidate.type == "model": + elif candidate.type == "model": generations = await self._run_model_generation(input_rows, task_config) + else: + raise ValueError(f"Invalid candidate type: {candidate.type}") # scoring with generated_answer score_input_rows = [