fix arg name

aws · Oct 26, 2020 · 037c25d · 037c25d
1 parent 82bff90
commit 037c25d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/integration/local/test_dummy.py b/test/integration/local/test_dummy.py
@@ -37,7 +37,7 @@ def container():
 
 def test_install_requirements(capsys):
     estimator = Estimator(
-        image_name="sagemaker-training-toolkit-test:dummy",
+        image_uri="sagemaker-training-toolkit-test:dummy",
         role="SageMakerRole",
         train_instance_count=1,
         train_instance_type="local",