From 4ad151f3208f274fba012f7c4adb845e05a12b9b Mon Sep 17 00:00:00 2001
From: Philip Paquette <pcpaquette@gmail.com>
Date: Sat, 11 Mar 2017 13:44:40 -0500
Subject: [PATCH] 0.0.6 - Max_episode_steps instead of timestep_limit

---
 .gitignore                                | 1 +
 ppaquette_gym_super_mario/__init__.py     | 4 ++--
 ppaquette_gym_super_mario/package_info.py | 2 +-
 requirements.txt                          | 2 +-
 setup.py                                  | 2 +-
 5 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/.gitignore b/.gitignore
index 5d61641..86fc5c5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -9,6 +9,7 @@
 
 # Virtualenv
 /env
+.env*
 
 # Python egg metadata, regenerated from source files by setuptools.
 /*.egg-info
diff --git a/ppaquette_gym_super_mario/__init__.py b/ppaquette_gym_super_mario/__init__.py
index 94931c8..a1f3e1e 100644
--- a/ppaquette_gym_super_mario/__init__.py
+++ b/ppaquette_gym_super_mario/__init__.py
@@ -23,7 +23,7 @@
     register(
         id='{}/meta-SuperMarioBros{}-v0'.format(USERNAME, tile_suffix),
         entry_point='{}_gym_super_mario:MetaSuperMarioBrosEnv'.format(USERNAME),
-        timestep_limit=9999999,
+        max_episode_steps=9999999,
         reward_threshold=32000,
         kwargs={ 'draw_tiles': draw_tiles, 'average_over': 3, 'passing_grade': 600, 'min_tries_for_avg': 3 },
         nondeterministic=True,
@@ -34,7 +34,7 @@
         register(
             id='{}/SuperMarioBros-{}-{}{}-v0'.format(USERNAME, world_number, level_number, tile_suffix),
             entry_point='{}_gym_super_mario:SuperMarioBrosEnv'.format(USERNAME),
-            timestep_limit=10000,
+            max_episode_steps=10000,
             reward_threshold=(max_distance - 40),
             kwargs={ 'draw_tiles': draw_tiles, 'level': level },
             # Seems to be non-deterministic about 5% of the time
diff --git a/ppaquette_gym_super_mario/package_info.py b/ppaquette_gym_super_mario/package_info.py
index dc9a48a..b10c0a8 100644
--- a/ppaquette_gym_super_mario/package_info.py
+++ b/ppaquette_gym_super_mario/package_info.py
@@ -1,2 +1,2 @@
-VERSION='0.0.5'
+VERSION='0.0.6'
 USERNAME='ppaquette'
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 39c0561..4551686 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1 @@
-gym>=0.2.3
+gym>=0.8.0
diff --git a/setup.py b/setup.py
index 5c46864..f1a87bc 100644
--- a/setup.py
+++ b/setup.py
@@ -17,5 +17,5 @@
     packages=[package for package in find_packages() if package.startswith(USERNAME)],
     package_data={ '{}_{}'.format(USERNAME, 'gym_super_mario'): ['lua/*.lua', 'roms/*.nes' ] },
     zip_safe=False,
-    install_requires=[ 'gym>=0.2.3' ],
+    install_requires=[ 'gym>=0.8.0' ],
 )