Skip to content

Commit 50a50c1

Browse files
author
Jaya Kasiraj
committed
fix: update unit tests after python changes
1 parent b7bfa3a commit 50a50c1

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tests/unit/sagemaker/workflow/test_steps.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -671,7 +671,7 @@ def test_processing_step_normalizes_args_with_local_code(mock_normalize_args, sc
671671
mock_normalize_args.return_value = [step.inputs, step.outputs]
672672
step.to_request()
673673
mock_normalize_args.assert_called_with(
674-
job_name="MyProcessingStep-a22fc59b38f13da26f6a40b18687ba598cf669f74104b793cefd9c63eddf4ac7",
674+
job_name=None,
675675
arguments=step.job_arguments,
676676
inputs=step.inputs,
677677
outputs=step.outputs,

tests/unit/sagemaker/workflow/test_utils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -80,11 +80,11 @@ def test_repack_model_step(estimator):
8080
assert hyperparameters["inference_script"] == '"dummy_script.py"'
8181
assert hyperparameters["model_archive"] == '"s3://my-bucket/model.tar.gz"'
8282
assert hyperparameters["sagemaker_program"] == f'"{REPACK_SCRIPT_LAUNCHER}"'
83-
assert (
84-
hyperparameters["sagemaker_submit_directory"]
85-
== '"s3://my-bucket/MyRepackModelStep-717d7bdd388168c27e9ad2938ff0314e35be50b3157cf2498688c7525ea27e1e\
86-
/source/sourcedir.tar.gz"'
87-
)
83+
84+
# ex: "gits3://my-bucket/sagemaker-scikit-learn-2025-04-07-20-39-38-854/source/sourcedir.tar.gz"
85+
sagemaker_submit_directory = hyperparameters["sagemaker_submit_directory"]
86+
assert sagemaker_submit_directory.startswith("\"s3://my-bucket/sagemaker-scikit-learn-")
87+
assert sagemaker_submit_directory.endswith("/source/sourcedir.tar.gz\"")
8888

8989
del request_dict["Arguments"]["HyperParameters"]
9090
del request_dict["Arguments"]["AlgorithmSpecification"]["TrainingImage"]

0 commit comments

Comments
 (0)