create script

Kautenja · May 8, 2019 · 9471aa7 · 9471aa7
1 parent c5c1f9a
commit 9471aa7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/scripts/run.py b/scripts/run.py
@@ -0,0 +1,15 @@
+from nes_py import NESEnv
+import tqdm
+env = NESEnv('./nes_py/tests/games/super-mario-bros-1.nes')
+
+done = True
+
+try:
+    for _ in tqdm.tqdm(range(5000)):
+        if done:
+            state = env.reset()
+            done = False
+        else:
+            state, reward, done, info = env.step(env.action_space.sample())
+except KeyboardInterrupt:
+    pass