Skip to content

Commit 8118a81

Browse files
author
Payton Staub
committed
fix: Remove sagemaker_job_name from hyperparameters in TrainingStep
1 parent 086258d commit 8118a81

File tree

3 files changed

+1
-997
lines changed

3 files changed

+1
-997
lines changed

src/sagemaker/workflow/steps.py

+1
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,7 @@ def arguments(self) -> RequestType:
301301
)
302302
request_dict = self.estimator.sagemaker_session._get_train_request(**train_args)
303303
request_dict.pop("TrainingJobName")
304+
request_dict["HyperParameters"].pop("sagemaker_job_name", None)
304305

305306
return request_dict
306307

0 commit comments

Comments
 (0)