fix naming for consistency

README.md

@@ -133,9 +133,9 @@ The random stage selection environment randomly selects a stage and allows a
 single attempt to clear it. Upon a death and subsequent call to `reset`, the
 environment randomly selects a new stage.  This is only available for the
 standard Super Mario Bros. game, _not_ Lost Levels (at the moment). To use
-these environments, append `RandomLevels` to the `SuperMarioBros` id. For
+these environments, append `RandomStages` to the `SuperMarioBros` id. For
 example, to use the standard ROM with random stage selection use
-`SuperMarioBrosRandomLevels-v0`. To seed the random stage selection use the
+`SuperMarioBrosRandomStages-v0`. To seed the random stage selection use the
 `seed` method of the env, i.e., `env.seed(1)`, before any calls to `reset`.
 
 ## Step

gym_super_mario_bros/__init__.py

@@ -1,12 +1,12 @@
 """Registration code of Gym environments in this package."""
 from .smb_env import SuperMarioBrosEnv
-from .smb_random_levels_env import SuperMarioBrosRandomLevelsEnv
+from .smb_random_stages_env import SuperMarioBrosRandomStagesEnv
 from ._registration import make
 
 
 # define the outward facing API of this package
 __all__ = [
     make.__name__,
     SuperMarioBrosEnv.__name__,
-    SuperMarioBrosRandomLevelsEnv.__name__,
+    SuperMarioBrosRandomStagesEnv.__name__,
 ]

gym_super_mario_bros/_registration.py

@@ -18,7 +18,7 @@ def _register_mario_env(id, is_random=False, **kwargs):
     # if the is random flag is set
     if is_random:
         # set the entry point to the random level environment
-        entry_point = 'gym_super_mario_bros:SuperMarioBrosRandomLevelsEnv'
+        entry_point = 'gym_super_mario_bros:SuperMarioBrosRandomStagesEnv'
     else:
         # set the entry point to the standard Super Mario Bros. environment
         entry_point = 'gym_super_mario_bros:SuperMarioBrosEnv'
@@ -41,10 +41,10 @@ def _register_mario_env(id, is_random=False, **kwargs):
 
 
 # Super Mario Bros. Random Levels
-_register_mario_env('SuperMarioBrosRandomLevels-v0', is_random=True, rom_mode='vanilla')
-_register_mario_env('SuperMarioBrosRandomLevels-v1', is_random=True, rom_mode='downsample')
-_register_mario_env('SuperMarioBrosRandomLevels-v2', is_random=True, rom_mode='pixel')
-_register_mario_env('SuperMarioBrosRandomLevels-v3', is_random=True, rom_mode='rectangle')
+_register_mario_env('SuperMarioBrosRandomStages-v0', is_random=True, rom_mode='vanilla')
+_register_mario_env('SuperMarioBrosRandomStages-v1', is_random=True, rom_mode='downsample')
+_register_mario_env('SuperMarioBrosRandomStages-v2', is_random=True, rom_mode='pixel')
+_register_mario_env('SuperMarioBrosRandomStages-v3', is_random=True, rom_mode='rectangle')
 
 
 # Super Mario Bros. 2 (Lost Levels)

gym_super_mario_bros/smb_random_levels_env.py → gym_super_mario_bros/smb_random_stages_env.py

@@ -5,7 +5,7 @@
 from .smb_env import SuperMarioBrosEnv
 
 
-class SuperMarioBrosRandomLevelsEnv(gym.Env):
+class SuperMarioBrosRandomStagesEnv(gym.Env):
     """A Super Mario Bros. environment that randomly selects levels."""
 
     # relevant meta-data about the environment
@@ -162,4 +162,4 @@ def get_keys_to_action(self):
 
 
 # explicitly define the outward facing API of this module
-__all__ = [SuperMarioBrosRandomLevelsEnv.__name__]
+__all__ = [SuperMarioBrosRandomStagesEnv.__name__]

gym_super_mario_bros/tests/test_registration.py

@@ -55,7 +55,7 @@ class ShouldMakeSuperMarioBros(ShouldMakeEnv, TestCase):
     env_id = ['SuperMarioBros-v{}'.format(v) for v in range(4)]
 
 
-class ShouldMakeSuperMarioBrosRandomLevels(ShouldMakeEnv, TestCase):
+class ShouldMakeSuperMarioBrosRandomStages(ShouldMakeEnv, TestCase):
     # the random number seed for this environment
     seed = 1
     # the amount of time left
@@ -65,7 +65,7 @@ class ShouldMakeSuperMarioBrosRandomLevels(ShouldMakeEnv, TestCase):
     # the current stage
     stage = 4
     # the environments ID for all versions of Super Mario Bros
-    env_id = ['SuperMarioBrosRandomLevels-v{}'.format(v) for v in range(4)]
+    env_id = ['SuperMarioBrosRandomStages-v{}'.format(v) for v in range(4)]
 
 
 class ShouldMakeSuperMarioBrosLostLevels(ShouldMakeEnv, TestCase):