reformatting

Author: icywang86rui

tests/integ/test_local_mode.py

@@ -92,7 +92,7 @@ def test_tf_local_mode(sagemaker_local_session):
         estimator = TensorFlow(
             entry_point=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             training_steps=1,
             evaluation_steps=1,
             hyperparameters={"input_tensor_name": "inputs"},
@@ -135,7 +135,7 @@ def test_tf_distributed_local_mode(sagemaker_local_session):
         estimator = TensorFlow(
             entry_point=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             training_steps=1,
             evaluation_steps=1,
             hyperparameters={"input_tensor_name": "inputs"},
@@ -177,7 +177,7 @@ def test_tf_local_data(sagemaker_local_session):
         estimator = TensorFlow(
             entry_point=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             training_steps=1,
             evaluation_steps=1,
             hyperparameters={"input_tensor_name": "inputs"},
@@ -218,7 +218,7 @@ def test_tf_local_data_local_script():
         estimator = TensorFlow(
             entry_point=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             training_steps=1,
             evaluation_steps=1,
             hyperparameters={"input_tensor_name": "inputs"},

tests/integ/test_tf_cifar.py

@@ -54,7 +54,7 @@ def test_cifar(sagemaker_session):
             entry_point="resnet_cifar_10.py",
             source_dir=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             training_steps=50,
             evaluation_steps=5,
             train_instance_count=2,

tests/integ/test_tf_keras.py

@@ -41,7 +41,7 @@ def test_keras(sagemaker_session):
             entry_point="keras_cnn_cifar_10.py",
             source_dir=script_path,
             role="SageMakerRole",
-            framework_version='1.12',
+            framework_version="1.12",
             sagemaker_session=sagemaker_session,
             hyperparameters={"learning_rate": 1e-4, "decay": 1e-6},
             training_steps=50,