Minor change

pytorch_rl/envs.py

@@ -6,7 +6,6 @@ from gym.spaces.box import Box
 try:
     import gym_minigrid
     from gym_minigrid.wrappers import *
-    #from gym_minigrid.envs import *
 except:
     pass
 

run_tests.py

@@ -6,9 +6,12 @@ import numpy as np
 from gym_minigrid.register import envSet
 from gym_minigrid.minigrid import Grid
 
-# Make sure directly importing a specific environment works
+# Test specifically importing a specific environment
 from gym_minigrid.envs import DoorKeyEnv
 
+# Test importing wrappers
+from gym_minigrid.wrappers import *
+
 print('%d environments registered' % len(envSet))
 
 for envName in sorted(envSet):