Added 5x5 config for the fetch environment

README.md

@@ -156,7 +156,8 @@ rooms and building a curriculum, the environment can be solved.
 ### Fetch environment
 
 Registered configurations:
-- `MiniGrid-Fetch-8x8-v0`
+- `MiniGrid-Fetch-5x5-N2-v0`
+- `MiniGrid-Fetch-8x8-N3-v0`
 
 <p align="center">
 <img src="/figures/fetch-env.png" width=392 height=269>

gym_minigrid/envs/fetch.py

@@ -109,7 +109,16 @@ class FetchEnv(MiniGridEnv):
 
         return obs, reward, done, info
 
+class FetchEnv5x5N2(FetchEnv):
+    def __init__(self):
+        super().__init__(size=5, numObjs=2)
+
+register(
+    id='MiniGrid-Fetch-5x5-N2-v0',
+    entry_point='gym_minigrid.envs:FetchEnv5x5N2'
+)
+
 register(
-    id='MiniGrid-Fetch-8x8-v0',
+    id='MiniGrid-Fetch-8x8-N3-v0',
     entry_point='gym_minigrid.envs:FetchEnv'
 )