fix bug in test

ndodda-amazon · ndodda-amazon · commit 1716b874f3a1 · 2021-01-21T13:35:21.000-08:00
diff --git a/tests/unit/test_estimator.py b/tests/unit/test_estimator.py
@@ -3250,6 +3250,7 @@ def test_image_name_map(sagemaker_session):
         "my_script.py",
         image_name=IMAGE_URI,
         role=ROLE,
+        sagemaker_session=sagemaker_session,
         instance_count=INSTANCE_COUNT,
         instance_type=INSTANCE_TYPE,
     )

Original file line number	Diff line number	Diff line change
`@@ -3250,6 +3250,7 @@ def test_image_name_map(sagemaker_session):`
`3250`	`3250`	`"my_script.py",`
`3251`	`3251`	`image_name=IMAGE_URI,`
`3252`	`3252`	`role=ROLE,`
	`3253`	`+ sagemaker_session=sagemaker_session,`
`3253`	`3254`	`instance_count=INSTANCE_COUNT,`
`3254`	`3255`	`instance_type=INSTANCE_TYPE,`
`3255`	`3256`	`)`