fix engine name

Signed-off-by: lilacheden <[email protected]>
IBM · Jan 16, 2025 · 14446cb · 14446cb
1 parent 225990b
commit 14446cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prepare/metrics/llm_as_judge/rag_judge.py b/prepare/metrics/llm_as_judge/rag_judge.py
@@ -102,7 +102,7 @@ def get_prediction_field(metric_type):
     "llama_3_3_70b_instruct_wml": "engines.classification.llama_3_3_70b_instruct_wml",
     "llama_3_3_70b_instruct_watsonx": "engines.classification.llama_3_3_70b_instruct_watsonx",
     "llama_3_3_70b_instruct_rits": "engines.classification.llama_3_3_70b_instruct_rits",
-    "gpt_4o_azure": "engines.classification.gpt_4o_azure",
+    "gpt_4o_azure": "engines.classification.gpt_4o_2024_08_06_azure_openai",
     generic_engine_label: GenericInferenceEngine(),
 }