Skip to content

Commit 4e37cf1

Browse files
author
Purak
committed
fix: Normalizing job_name in the ProcessingStep.
1 parent 99f023e commit 4e37cf1

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/sagemaker/workflow/steps.py

+2
Original file line numberDiff line numberDiff line change
@@ -496,6 +496,7 @@ def __init__(
496496
self.job_arguments = job_arguments
497497
self.code = code
498498
self.property_files = property_files
499+
self.job_name = name
499500

500501
# Examine why run method in sagemaker.processing.Processor mutates the processor instance
501502
# by setting the instance's arguments attribute. Refactor Processor.run, if possible.
@@ -514,6 +515,7 @@ def arguments(self) -> RequestType:
514515
ProcessingJobName and ExperimentConfig cannot be included in the arguments.
515516
"""
516517
normalized_inputs, normalized_outputs = self.processor._normalize_args(
518+
job_name=self.job_name,
517519
arguments=self.job_arguments,
518520
inputs=self.inputs,
519521
outputs=self.outputs,

tests/unit/sagemaker/workflow/test_steps.py

+1
Original file line numberDiff line numberDiff line change
@@ -520,6 +520,7 @@ def test_processing_step_normalizes_args(mock_normalize_args, sagemaker_session)
520520
mock_normalize_args.return_value = [step.inputs, step.outputs]
521521
step.to_request()
522522
mock_normalize_args.assert_called_with(
523+
job_name=step.name,
523524
arguments=step.job_arguments,
524525
inputs=step.inputs,
525526
outputs=step.outputs,

0 commit comments

Comments
 (0)