Update logging information for submitting and deleting training job (#189)

Co-authored-by: pintaoz <[email protected]>

Author: pintaoz-aws

src/sagemaker/hyperpod/cli/commands/training.py

@@ -1,16 +1,7 @@
 import click
-import logging
-import os
-import yaml
-import shutil
-import subprocess
-from pathlib import Path
 from sagemaker.hyperpod.training.hyperpod_pytorch_job import HyperPodPytorchJob
 from sagemaker.hyperpod.common.config import Metadata
-import tempfile
-from typing import List, Dict, Any, Optional, Callable, get_args, get_origin, Literal
 from sagemaker.hyperpod.cli.training_utils import generate_click_command
-from importlib.metadata import entry_points
 from hyperpod_pytorch_job_template.registry import SCHEMA_REGISTRY
 from sagemaker.hyperpod.common.telemetry.telemetry_logging import (
     _hyperpod_telemetry_emitter,

src/sagemaker/hyperpod/training/hyperpod_pytorch_job.py

@@ -3,8 +3,8 @@
     _HyperPodPytorchJob, HyperPodPytorchJobStatus
 )
 from sagemaker.hyperpod.common.config.metadata import Metadata
-from kubernetes import client, config, __version__ as kubernetes_client_version
-from typing import List, Optional, ClassVar, Tuple
+from kubernetes import client, config
+from typing import List, Optional, ClassVar
 from sagemaker.hyperpod.common.utils import (
     handle_exception,
     get_default_namespace,
@@ -84,7 +84,7 @@ def create(self, debug=False):
                 plural=PLURAL,
                 body=config,
             )
-            logger.info("Successfully submitted HyperPodPytorchJob!")
+            logger.info(f"Successfully submitted HyperPodPytorchJob '{self.metadata.name}'!")
         except Exception as e:
             logger.error(f"Failed to create HyperPodPytorchJob {self.metadata.name}!")
             handle_exception(e, self.metadata.name, self.metadata.namespace)
@@ -131,7 +131,7 @@ def delete(self):
                 plural=PLURAL,
                 name=self.metadata.name,
             )
-            logger.info(f"Successful deleted HyperPodPytorchJob!")
+            logger.info(f"Successful deleted HyperPodPytorchJob '{self.metadata.name}'!")
         except Exception as e:
             logger.error(f"Failed to delete HyperPodPytorchJob {self.metadata.name}!")
             handle_exception(e, self.metadata.name, self.metadata.namespace)