Fix unit tests

Payton Staub · Payton Staub · commit 13261816685d · 2021-09-17T17:18:19.000-07:00
diff --git a/tests/unit/sagemaker/workflow/test_step_collections.py b/tests/unit/sagemaker/workflow/test_step_collections.py
@@ -532,6 +532,8 @@ def test_register_model_with_model_repack_with_model(model, model_metrics):
                         "sagemaker_container_log_level": "20",
                         "sagemaker_job_name": repacker_job_name,
                         "sagemaker_region": f'"{REGION}"',
+                        "dependencies": 'null',
+                        "source_dir": 'null'
                     },
                     "InputDataConfig": [
                         {

Original file line number	Diff line number	Diff line change
`@@ -532,6 +532,8 @@ def test_register_model_with_model_repack_with_model(model, model_metrics):`
`532`	`532`	`"sagemaker_container_log_level": "20",`
`533`	`533`	`"sagemaker_job_name": repacker_job_name,`
`534`	`534`	`"sagemaker_region": f'"{REGION}"',`
	`535`	`+ "dependencies": 'null',`
	`536`	`+ "source_dir": 'null'`
`535`	`537`	`},`
`536`	`538`	`"InputDataConfig": [`
`537`	`539`	`{`