Fix Keras test (aws#132)

This test is only configured to run with 'local'. Change it to use the correct instance type accordingly.

Author: icywang86rui

test/integration/local/test_keras.py

@@ -35,7 +35,7 @@ def test_keras_training(sagemaker_local_session, docker_image, tmpdir, framework
         entry_point=entry_point,
         role='SageMakerRole',
         train_instance_count=1,
-        train_instance_type='local',
+        train_instance_type=local_mode_instance_type,
         image_name=docker_image,
         sagemaker_session=sagemaker_local_session,
         model_dir='/opt/ml/model',