diff --git a/sagemaker-train/src/sagemaker/train/common_utils/show_results_utils.py b/sagemaker-train/src/sagemaker/train/common_utils/show_results_utils.py index 55d478a730..02d56913fd 100644 --- a/sagemaker-train/src/sagemaker/train/common_utils/show_results_utils.py +++ b/sagemaker-train/src/sagemaker/train/common_utils/show_results_utils.py @@ -677,6 +677,11 @@ def _show_llmaj_results( pipeline_execution, "EvaluateBaseModelMetrics" ) + if not custom_job_name and not base_job_name: + custom_job_name = _extract_training_job_name_from_steps( + pipeline_execution, "EvaluateWithJudge" + ) + # Handle single-model evaluation scenarios if not custom_job_name and not base_job_name: raise ValueError(