Reorganized files

README.md

@@ -69,7 +69,7 @@ python3 basicrl/enjoy.py --env-name MiniGrid-Empty-8x8-v0 --load-dir ./trained_m
 
 ## Included Environments
 
-The environments listed below are implemented and registered in [simple_envs.py](gym_minigrid/envs/simple_envs.py).
+The environments listed below are implemented in the [gym_minigrid/envs](/gym_minigrid/envs) directory.
 
 ### Empty environment
 

gym_minigrid/__init__.py

@@ -1 +1,2 @@
+# Import these environments to make sure they register themselves
 import gym_minigrid.envs.simple_envs

gym_minigrid/envs/__init__.py

@@ -1,2 +1,2 @@
-from gym_minigrid.envs.minigrid_env import MiniGridEnv
+from gym_minigrid.envs.minigrid import MiniGridEnv
 from gym_minigrid.envs.simple_envs import *

gym_minigrid/envs/minigrid_env.py

@@ -3,7 +3,7 @@ import gym
 from gym import error, spaces, utils
 from gym.utils import seeding
 import numpy as np
-from gym_minigrid.envs.rendering import *
+from gym_minigrid.rendering import *
 
 # Size in pixels of a cell in the full-scale human view
 CELL_PIXELS = 32

gym_minigrid/envs/simple_envs.py

@@ -1,5 +1,5 @@
-from gym_minigrid.envs.minigrid_env import *
-from gym_minigrid.envs.register import register
+from gym_minigrid.envs.minigrid import *
+from gym_minigrid.register import register
 
 class EmptyEnv(MiniGridEnv):
     """

run_tests.py

@@ -2,7 +2,7 @@
 
 import random
 import gym
-from gym_minigrid.envs.register import envSet
+from gym_minigrid.register import envSet
 
 print('%d environments registered' % len(envSet))