From 28e69162daee39f0c5cdb6d37f3a385b36a5a70b Mon Sep 17 00:00:00 2001
From: pythonlessons <43852392+pythonlessons@users.noreply.github.com>
Date: Mon, 23 Mar 2020 10:36:30 +0200
Subject: [PATCH] Update Cartpole_DQN.py

---
 01_CartPole-reinforcement-learning/Cartpole_DQN.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/01_CartPole-reinforcement-learning/Cartpole_DQN.py b/01_CartPole-reinforcement-learning/Cartpole_DQN.py
index 4c4d939..c8178f5 100644
--- a/01_CartPole-reinforcement-learning/Cartpole_DQN.py
+++ b/01_CartPole-reinforcement-learning/Cartpole_DQN.py
@@ -1,3 +1,6 @@
+# Tutorial by www.pylessons.com
+# Tutorial written for - Tensorflow 1.15, Keras 2.2.4
+
 import os
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 import random
@@ -153,4 +156,4 @@ def test(self):
 if __name__ == "__main__":
     agent = DQNAgent()
     #agent.run()
-    agent.test()
\ No newline at end of file
+    agent.test()