diff --git a/sagemaker-train/src/sagemaker/train/evaluate/execution.py b/sagemaker-train/src/sagemaker/train/evaluate/execution.py index e2388ef313..05008681ec 100644 --- a/sagemaker-train/src/sagemaker/train/evaluate/execution.py +++ b/sagemaker-train/src/sagemaker/train/evaluate/execution.py @@ -904,7 +904,7 @@ def wait( if ipython is not None and 'IPKernelApp' in ipython.config: is_jupyter = True from IPython.display import display, HTML, clear_output - except: + except Exception: pass if is_jupyter: @@ -958,7 +958,7 @@ def wait( end = datetime.fromisoformat(step.end_time.replace('Z', '+00:00')) duration_seconds = (end - start).total_seconds() duration = f"{duration_seconds:.1f}s" - except: + except Exception: duration = "N/A" elif step.start_time: duration = "Running..." @@ -982,7 +982,7 @@ def wait( # Add error indicator if failures exist if has_failures: if step.failure_reason: - row_data.append("❌") + row_data.append("❌") failed_steps.append(step) else: row_data.append("") @@ -1005,7 +1005,7 @@ def wait( for step in failed_steps: content_parts.append(Text("")) # Empty line before each failure - content_parts.append(Text(f"• {step.display_name or step.name}:", style="bold red")) + content_parts.append(Text(f"• {step.display_name or step.name}:", style="bold red")) content_parts.append(Text(f" {step.failure_reason}", style="red")) combined_content = Group(*content_parts)