From 6de3421b2ee477ab1bf2104fe6cdaf7b3616cb34 Mon Sep 17 00:00:00 2001
From: Siju Samuel <siju.samuel@huawei.com>
Date: Wed, 10 Jul 2019 14:42:30 +0530
Subject: [PATCH] Deprecated tf.session removed in mnist_tflite.py

---
 tensorflow/lite/tutorials/mnist_tflite.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/lite/tutorials/mnist_tflite.py b/tensorflow/lite/tutorials/mnist_tflite.py
index 6cc58461635..6434277f096 100644
--- a/tensorflow/lite/tutorials/mnist_tflite.py
+++ b/tensorflow/lite/tutorials/mnist_tflite.py
@@ -33,7 +33,7 @@ flags = flags.FLAGS
 
 def test_image_generator():
   # Generates an iterator over images
-  with tf.Session() as sess:
+  with tf.compat.v1.Session() as sess:
     input_data = tf.compat.v1.data.make_one_shot_iterator(dataset.test(
         flags.data_dir)).get_next()
     try: