From 90f1f68840e247258d4164c6c089b18969aeb4c4 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Mon, 2 Oct 2023 12:08:10 -0700 Subject: [PATCH] fix: transform step unit test --- tests/unit/sagemaker/workflow/test_transform_step.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/sagemaker/workflow/test_transform_step.py b/tests/unit/sagemaker/workflow/test_transform_step.py index 856c6d8324..d22965dae8 100644 --- a/tests/unit/sagemaker/workflow/test_transform_step.py +++ b/tests/unit/sagemaker/workflow/test_transform_step.py @@ -127,7 +127,7 @@ def test_transform_step_with_transformer(model_name, data, output_path, pipeline ) step_def = json.loads(pipeline.definition())["Steps"][0] - assert step_def["Arguments"]["TransformJobName"] == "TestTransformJobPrefix" + assert step_def["Arguments"]["TransformJobName"].startswith("TestTransformJobPrefix") assert step_def == { "Name": "MyTransformStep", "Type": "Transform",