Skip to content

Commit 76c4e32

Browse files
committed
Let evaluator name to be a string
Handle case were self.inference_engine.get_engine_id() is not implemented Signed-off-by: Martín Santillán Cooper <[email protected]>
1 parent 54188d1 commit 76c4e32

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/unitxt/llm_as_judge.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ class LLMJudge(BulkInstanceMetric):
5757
# option_selection_strategy: OptionSelectionStrategyEnum = (
5858
# OptionSelectionStrategyEnum.PARSE_OUTPUT_TEXT
5959
# )
60-
evaluator_name: EvaluatorNameEnum = None
60+
evaluator_name: Optional[Union[str,EvaluatorNameEnum]] = None
6161
check_positional_bias: bool = True
6262
context_fields: Union[str, List[str], Dict[str, str]] = ["context"]
6363
generate_summaries: bool = True
@@ -77,9 +77,10 @@ def prepare(self):
7777
}
7878

7979
if self.evaluator_name is None:
80-
self.evaluator_name = self.inference_engine.get_engine_id()
81-
elif not isinstance(self.evaluator_name, EvaluatorNameEnum):
82-
self.evaluator_name = EvaluatorNameEnum[self.evaluator_name]
80+
try:
81+
self.evaluator_name = self.inference_engine.get_engine_id()
82+
except NotImplementedError:
83+
self.evaluator_name = "not_set"
8384

8485
def before_process_multi_stream(self):
8586
super().before_process_multi_stream()

0 commit comments

Comments
 (0)