From 037c25d4dc48f9187581adba441c2177b91001e1 Mon Sep 17 00:00:00 2001
From: Chuyang Deng <chuyangd@amazon.com>
Date: Mon, 26 Oct 2020 11:11:39 -0700
Subject: [PATCH] fix arg name

---
 test/integration/local/test_dummy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/integration/local/test_dummy.py b/test/integration/local/test_dummy.py
index 8c1342e2..7ba09999 100644
--- 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",