diff --git a/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py b/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py index 3da9dc95..1bafa76f 100644 --- a/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py +++ b/hyperpod-pytorch-job-template/hyperpod_pytorch_job_template/v1_0/model.py @@ -353,6 +353,7 @@ def to_domain(self) -> Dict: result = { "name": self.job_name, "namespace": self.namespace, + "labels": metadata_labels, "spec": job_kwargs, } return result diff --git a/src/sagemaker/hyperpod/cli/commands/training.py b/src/sagemaker/hyperpod/cli/commands/training.py index 25688902..15978f9b 100644 --- a/src/sagemaker/hyperpod/cli/commands/training.py +++ b/src/sagemaker/hyperpod/cli/commands/training.py @@ -33,11 +33,14 @@ def pytorch_create(version, debug, config): job_name = config.get("name") namespace = config.get("namespace") spec = config.get("spec") + metadata_labels = config.get("labels") # Prepare metadata metadata_kwargs = {"name": job_name} if namespace: metadata_kwargs["namespace"] = namespace + if metadata_labels: + metadata_kwargs["labels"] = metadata_labels # Prepare job kwargs job_kwargs = {