Skip to content

Commit 7166342

Browse files
author
Ignacio Quintero
committed
Fix integ test
1 parent 196c9fe commit 7166342

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_mxnet_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def test_async_fit(sagemaker_session, mxnet_full_version):
8181
mx = MXNet(entry_point=script_path, role='SageMakerRole', py_version=PYTHON_VERSION,
8282
train_instance_count=1, train_instance_type='ml.c4.xlarge',
8383
sagemaker_session=sagemaker_session, framework_version=mxnet_full_version,
84-
distributions={'parameter_server': {'enabled': True}})
84+
distribution={'parameter_server': {'enabled': True}})
8585

8686
train_input = mx.sagemaker_session.upload_data(path=os.path.join(data_path, 'train'),
8787
key_prefix='integ-test-data/mxnet_mnist/train')

0 commit comments

Comments
 (0)