Update all MXNet integ tests to use the latest version (#589)

Author: laurenyu

tests/data/mxnet_mnist/mnist_framework_mode.py

@@ -286,12 +286,13 @@ def test_mxnet_local_mode(sagemaker_local_session, mxnet_full_version):
             mx.delete_endpoint()
 
 
-def test_mxnet_local_data_local_script():
+def test_mxnet_local_data_local_script(mxnet_full_version):
     data_path = os.path.join(DATA_DIR, 'mxnet_mnist')
-    script_path = os.path.join(data_path, 'mnist_framework_mode.py')
+    script_path = os.path.join(data_path, 'mnist.py')
 
     mx = MXNet(entry_point=script_path, role='SageMakerRole',
                train_instance_count=1, train_instance_type='local',
+               framework_version=mxnet_full_version,
                sagemaker_session=LocalNoS3Session())
 
     train_input = 'file://' + os.path.join(data_path, 'train')

tests/integ/test_local_mode.py

@@ -404,20 +404,20 @@ def test_stop_tuning_job(sagemaker_session):
 
 
 @pytest.mark.continuous_testing
-def test_tuning_mxnet(sagemaker_session):
+def test_tuning_mxnet(sagemaker_session, mxnet_full_version):
     with timeout(minutes=TUNING_DEFAULT_TIMEOUT_MINUTES):
-        script_path = os.path.join(DATA_DIR, 'mxnet_mnist', 'mnist_framework_mode.py')
+        script_path = os.path.join(DATA_DIR, 'mxnet_mnist', 'mnist.py')
         data_path = os.path.join(DATA_DIR, 'mxnet_mnist')
 
         estimator = MXNet(entry_point=script_path,
                           role='SageMakerRole',
                           py_version=PYTHON_VERSION,
                           train_instance_count=1,
                           train_instance_type='ml.m4.xlarge',
-                          framework_version='1.2.1',
+                          framework_version=mxnet_full_version,
                           sagemaker_session=sagemaker_session)
 
-        hyperparameter_ranges = {'learning_rate': ContinuousParameter(0.01, 0.2)}
+        hyperparameter_ranges = {'learning-rate': ContinuousParameter(0.01, 0.2)}
         objective_metric_name = 'Validation-accuracy'
         metric_definitions = [
             {'Name': 'Validation-accuracy', 'Regex': 'Validation-accuracy=([0-9\\.]+)'}]