Pass sagemaker_session to estimator

Author: yangaws

tests/integ/test_tf_keras.py

@@ -30,7 +30,7 @@ def test_keras(sagemaker_session, tf_full_version):
     with timeout(minutes=45):
         estimator = TensorFlow(entry_point='keras_cnn_cifar_10.py',
                                source_dir=script_path,
-                               role='SageMakerRole',
+                               role='SageMakerRole', sagemaker_session=sagemaker_session,
                                hyperparameters={'learning_rate': 1e-4, 'decay': 1e-6},
                                training_steps=500, evaluation_steps=5,
                                train_instance_count=1, train_instance_type='ml.c4.xlarge',