From 0fd59194421e56989d83116421aba75434849b82 Mon Sep 17 00:00:00 2001
From: Yu <yyumzn@amazon.com>
Date: Mon, 27 Aug 2018 13:42:20 -0700
Subject: [PATCH] Pass sagemaker_session to estimator

---
 tests/integ/test_tf_keras.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integ/test_tf_keras.py b/tests/integ/test_tf_keras.py
index 2fca1a81c8..37880dbe72 100644
--- a/tests/integ/test_tf_keras.py
+++ b/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',