From f315463bfcb6ed0ad39223a3fe6cee884ab6be97 Mon Sep 17 00:00:00 2001
From: Aneesh Panoli <aneeshpanoli@gmail.com>
Date: Mon, 22 Jun 2020 01:16:43 -0700
Subject: [PATCH] Fix for breaking changes in tensorflow v2.0

There are several module-level changes in TensorFlow v2, which breaks the notebook.  This is the recommended fix.
---
 DQN.ipynb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/DQN.ipynb b/DQN.ipynb
index d88f2c0..ab512c8 100755
--- a/DQN.ipynb
+++ b/DQN.ipynb
@@ -141,7 +141,8 @@
     "import os\n",
     "import random\n",
     "import gym\n",
-    "import tensorflow as tf\n",
+    "import tensorflow.compat.v1 as tf \n",
+    "tf.disable_v2_behavior() \n",
     "import numpy as np\n",
     "import imageio\n",
     "from skimage.transform import resize"