Skip to content

Commit 5fb9c0e

Browse files
authored
Update test_steps.py
1 parent 3f90257 commit 5fb9c0e

File tree

1 file changed

+0
-34
lines changed

1 file changed

+0
-34
lines changed

tests/unit/sagemaker/workflow/test_steps.py

Lines changed: 0 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -693,40 +693,6 @@ def test_processing_step_normalizes_args_with_no_code(mock_normalize_args, scrip
693693
)
694694

695695

696-
@patch("sagemaker.processing.ScriptProcessor._normalize_args")
697-
def test_processing_step_normalizes_args_with_no_code(mock_normalize_args, script_processor):
698-
cache_config = CacheConfig(enable_caching=True, expire_after="PT1H")
699-
inputs = [
700-
ProcessingInput(
701-
source=f"s3://{BUCKET}/processing_manifest",
702-
destination="processing_manifest",
703-
)
704-
]
705-
outputs = [
706-
ProcessingOutput(
707-
source=f"s3://{BUCKET}/processing_manifest",
708-
destination="processing_manifest",
709-
)
710-
]
711-
step = ProcessingStep(
712-
name="MyProcessingStep",
713-
processor=script_processor,
714-
inputs=inputs,
715-
outputs=outputs,
716-
job_arguments=["arg1", "arg2"],
717-
cache_config=cache_config,
718-
)
719-
mock_normalize_args.return_value = [step.inputs, step.outputs]
720-
step.to_request()
721-
mock_normalize_args.assert_called_with(
722-
job_name=None,
723-
arguments=step.job_arguments,
724-
inputs=step.inputs,
725-
outputs=step.outputs,
726-
code=None,
727-
)
728-
729-
730696
def test_create_model_step(sagemaker_session):
731697
model = Model(
732698
image_uri=IMAGE_URI,

0 commit comments

Comments
 (0)