Added missing import

IBM · Jan 24, 2025 · 6734429 · 6734429
1 parent 67355ef
commit 6734429
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/unitxt/llm_as_judge.py b/src/unitxt/llm_as_judge.py
@@ -8,6 +8,7 @@
 from .error_utils import UnitxtError
 from .inference import (
     InferenceEngine,
+    OptionSelectingByLogProbsInferenceEngine,
 )
 from .llm_as_judge_chat_templates import direct_template_dict, pairwise_template_dict
 from .llm_as_judge_constants import (