fix: transform step unit test

benieric · benieric · commit 90f1f68840e2 · 2023-10-02T12:08:10.000-07:00
diff --git 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",

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ def test_transform_step_with_transformer(model_name, data, output_path, pipeline`
`127`	`127`	`)`
`128`	`128`
`129`	`129`	`step_def = json.loads(pipeline.definition())["Steps"][0]`
`130`		`- assert step_def["Arguments"]["TransformJobName"] == "TestTransformJobPrefix"`
	`130`	`+ assert step_def["Arguments"]["TransformJobName"].startswith("TestTransformJobPrefix")`
`131`	`131`	`assert step_def == {`
`132`	`132`	`"Name": "MyTransformStep",`
`133`	`133`	`"Type": "Transform",`