diff --git a/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py b/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py
index f8d1d2e5b9..cd1e6274f3 100644
--- a/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py
+++ b/sagemaker-train/src/sagemaker/train/common_utils/finetune_utils.py
@@ -1192,8 +1192,7 @@ def _resolve_model_and_name(model, sagemaker_session=None):
         if not region_name:
             try:
                 import boto3
-
-                region_name = boto3.Session().region_name or os.environ.get("AWS_DEFAULT_REGION")
+                region_name = boto3.Session().region_name or os.environ.get('AWS_DEFAULT_REGION')
             except Exception:
                 pass
 
diff --git a/sagemaker-train/src/sagemaker/train/evaluate/execution.py b/sagemaker-train/src/sagemaker/train/evaluate/execution.py
index 04810ac97d..1395ecfa29 100644
--- a/sagemaker-train/src/sagemaker/train/evaluate/execution.py
+++ b/sagemaker-train/src/sagemaker/train/evaluate/execution.py
@@ -957,7 +957,7 @@ def wait(
             ipython = get_ipython()
             if ipython is not None and "IPKernelApp" in ipython.config:
                 is_jupyter = True
-                from IPython.display import clear_output
+                from IPython.display import display, HTML, clear_output
         except Exception:
             pass