Add property check

Payton Staub · Payton Staub · commit 2dfde229ff6f · 2022-03-02T16:04:30.000-08:00
diff --git a/src/sagemaker/workflow/steps.py b/src/sagemaker/workflow/steps.py
@@ -301,7 +301,8 @@ def arguments(self) -> RequestType:
         )
         request_dict = self.estimator.sagemaker_session._get_train_request(**train_args)
         request_dict.pop("TrainingJobName")
-        request_dict["HyperParameters"].pop("sagemaker_job_name", None)
+        if "HyperParameters" in request_dict:
+            request_dict["HyperParameters"].pop("sagemaker_job_name", None)
 
         return request_dict
 

Original file line number	Diff line number	Diff line change
`@@ -301,7 +301,8 @@ def arguments(self) -> RequestType:`
`301`	`301`	`)`
`302`	`302`	`request_dict = self.estimator.sagemaker_session._get_train_request(**train_args)`
`303`	`303`	`request_dict.pop("TrainingJobName")`
`304`		`- request_dict["HyperParameters"].pop("sagemaker_job_name", None)`
	`304`	`+ if "HyperParameters" in request_dict:`
	`305`	`+ request_dict["HyperParameters"].pop("sagemaker_job_name", None)`
`305`	`306`
`306`	`307`	`return request_dict`
`307`	`308`