fix integ tests

laurenyu · laurenyu · commit 06da1bc8c063 · 2020-07-01T16:31:41.000-07:00
diff --git a/tests/integ/test_horovod.py b/tests/integ/test_horovod.py
@@ -82,7 +82,7 @@ def test_horovod_local_mode(
         output_path=output_path,
         framework_version=tf_training_latest_version,
         py_version=tf_training_latest_py_version,
-        distributions={"mpi": {"enabled": True, "processes_per_host": processes}},
+        distribution={"mpi": {"enabled": True, "processes_per_host": processes}},
     )
 
     with timeout.timeout(minutes=integ.TRAINING_DEFAULT_TIMEOUT_MINUTES):
@@ -128,7 +128,7 @@ def _create_and_fit_estimator(sagemaker_session, tf_version, py_version, instanc
         sagemaker_session=sagemaker_session,
         py_version=py_version,
         framework_version=tf_version,
-        distributions={"mpi": {"enabled": True}},
+        distribution={"mpi": {"enabled": True}},
     )
 
     with timeout.timeout(minutes=integ.TRAINING_DEFAULT_TIMEOUT_MINUTES):
diff --git a/tests/integ/test_local_mode.py b/tests/integ/test_local_mode.py
@@ -169,7 +169,7 @@ def test_mxnet_distributed_local_mode(
         train_instance_type="local",
         sagemaker_session=sagemaker_local_session,
         framework_version=mxnet_full_version,
-        distributions={"parameter_server": {"enabled": True}},
+        distribution={"parameter_server": {"enabled": True}},
     )
 
     train_input = mx.sagemaker_session.upload_data(
diff --git a/tests/integ/test_mxnet_train.py b/tests/integ/test_mxnet_train.py
@@ -300,7 +300,7 @@ def test_async_fit(sagemaker_session, mxnet_full_version, mxnet_full_py_version,
             train_instance_type=cpu_instance_type,
             sagemaker_session=sagemaker_session,
             framework_version=mxnet_full_version,
-            distributions={"parameter_server": {"enabled": True}},
+            distribution={"parameter_server": {"enabled": True}},
         )
 
         train_input = mx.sagemaker_session.upload_data(
diff --git a/tests/integ/test_tf.py b/tests/integ/test_tf.py
@@ -134,7 +134,7 @@ def test_mnist_distributed(
         sagemaker_session=sagemaker_session,
         framework_version=tf_training_latest_version,
         py_version=tf_training_latest_py_version,
-        distributions=PARAMETER_SERVER_DISTRIBUTION,
+        distribution=PARAMETER_SERVER_DISTRIBUTION,
     )
     inputs = estimator.sagemaker_session.upload_data(
         path=os.path.join(MNIST_RESOURCE_PATH, "data"), key_prefix="scriptmode/distributed_mnist"

Original file line number	Diff line number	Diff line change
`@@ -169,7 +169,7 @@ def test_mxnet_distributed_local_mode(`
`169`	`169`	`train_instance_type="local",`
`170`	`170`	`sagemaker_session=sagemaker_local_session,`
`171`	`171`	`framework_version=mxnet_full_version,`
`172`		`- distributions={"parameter_server": {"enabled": True}},`
	`172`	`+ distribution={"parameter_server": {"enabled": True}},`
`173`	`173`	`)`
`174`	`174`
`175`	`175`	`train_input = mx.sagemaker_session.upload_data(`
Original file line number	Diff line number	Diff line change
`@@ -300,7 +300,7 @@ def test_async_fit(sagemaker_session, mxnet_full_version, mxnet_full_py_version,`
`300`	`300`	`train_instance_type=cpu_instance_type,`
`301`	`301`	`sagemaker_session=sagemaker_session,`
`302`	`302`	`framework_version=mxnet_full_version,`
`303`		`- distributions={"parameter_server": {"enabled": True}},`
	`303`	`+ distribution={"parameter_server": {"enabled": True}},`
`304`	`304`	`)`
`305`	`305`
`306`	`306`	`train_input = mx.sagemaker_session.upload_data(`
Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,7 @@ def test_mnist_distributed(`
`134`	`134`	`sagemaker_session=sagemaker_session,`
`135`	`135`	`framework_version=tf_training_latest_version,`
`136`	`136`	`py_version=tf_training_latest_py_version,`
`137`		`- distributions=PARAMETER_SERVER_DISTRIBUTION,`
	`137`	`+ distribution=PARAMETER_SERVER_DISTRIBUTION,`
`138`	`138`	`)`
`139`	`139`	`inputs = estimator.sagemaker_session.upload_data(`
`140`	`140`	`path=os.path.join(MNIST_RESOURCE_PATH, "data"), key_prefix="scriptmode/distributed_mnist"`