Added screenshots for KeyCorridor (#58)

* Added screenshots for KeyCorridor

* Reordered level id and evaluate, and displayed level ids as list

Author: lcswillems

README.md

@@ -128,8 +128,8 @@ A work-in-progress review of related work can be found [here](https://www.overle
 
 - Environment: The agent is placed in a room with a door.
 - Instruction: Open the red door
-- Level id: `BabyAI-OpenRedDoor-v0`
 - Evaluate: image understanding
+- Level id: `BabyAI-OpenRedDoor-v0`
 
 <p align="center"><img src="media/OpenRedDoor.png" width="250"></p>
 
@@ -140,35 +140,43 @@ A work-in-progress review of related work can be found [here](https://www.overle
     - a given color or location in `OpenDoor`
     - a given color in `OpenDoorColor`
     - a given location in `OpenDoorLoc`
-- Level id: `BabyAI-OpenDoor-v0`, `BabyAI-OpenDoorDebug-v0`, `BabyAI-OpenDoorColor-v0`, `BabyAI-OpenDoorColorDebug-v0`, `BabyAI-OpenDoorLoc-v0`, `BabyAI-OpenDoorLocDebug-v0`
 - Evaluate: image & text understanding, memory in `OpenDoor` and `OpenDoorLoc`
+- Level id:
+    - `BabyAI-OpenDoor-v0`
+    - `BabyAI-OpenDoorDebug-v0`
+    - `BabyAI-OpenDoorColor-v0`
+    - `BabyAI-OpenDoorColorDebug-v0`
+    - `BabyAI-OpenDoorLoc-v0`
+    - `BabyAI-OpenDoorLocDebug-v0`
 
 <p align="center"><img src="media/OpenDoor.png" width="250"></p>
 
 ### GoToObjDoor
 
 - Environment: The agent is placed in a room with 4 different doors and 5 different objects.
 - Instruction: Go to an object or a door of a given type and color
-- Level id: `BabyAI-GoToObjDoor-v0`
 - Evaluate: image & text understanding
+- Level id: `BabyAI-GoToObjDoor-v0`
 
 <p align="center"><img src="media/GoToObjDoor.png" width="250"></p>
 
 ### ActionObjDoor
 
 - Environment: The agent is placed in a room with 4 different doors and 5 different objects.
 - Instruction: [Pick up an object] or [go to an object or door] or [open a door]
-- Level id: `BabyAI-ActionObjDoor-v0`
 - Evaluate: image & text understanding
+- Level id: `BabyAI-ActionObjDoor-v0`
 
 <p align="center"><img src="media/ActionObjDoor.png" width="250"></p>
 
 ### Unlock
 
 - Environment: The agent is placed in a room with a key and a locked door. The room has either no distractors in `Unlock` or 3 distractors in `UnlockDist`.
 - Instruction: Open the door
-- Level id: `BabyAI-Unlock-v0`, `BabyAI-UnlockDist-v0`
 - Evaluate: image understanding, memory in `UnlockDist`
+- Level id:
+    - `BabyAI-Unlock-v0`
+    - `BabyAI-UnlockDist-v0`
 
 <p align="center">
     <img src="media/Unlock.png" width="250">
@@ -179,8 +187,8 @@ A work-in-progress review of related work can be found [here](https://www.overle
 
 - Environment: The agent is placed in a room with a key and a locked door. The door opens onto a room with a box. Rooms have either no distractors in `UnlockPickup` or 4 distractors in `UnlockPickupDist`.
 - Instruction: Pick up an object of a given type and color
-- Level id: `BabyAI-UnlockPickup-v0`, `BabyAI-UnlockPickupDist-v0`
 - Evaluate: image understanding, memory in `UnlockPickupDist`
+- Level id: `BabyAI-UnlockPickup-v0`, `BabyAI-UnlockPickupDist-v0`
 
 <p align="center">
     <img src="media/UnlockPickup.png" width="250">
@@ -191,71 +199,80 @@ A work-in-progress review of related work can be found [here](https://www.overle
 
 - Environment: The agent is placed in a room with a key and a locked door. The door is blocked by a ball. The door opens onto a room with a box.
 - Instruction: Pick up the box
-- Level id: `BabyAI-BlockedUnlockPickup-v0`
 - Evaluate: image understanding
+- Level id: `BabyAI-BlockedUnlockPickup-v0`
 
 <p align="center"><img src="media/BlockedUnlockPickup.png" width="250"></p>
 
 ### UnlockToUnlock
 
 - Environment: The agent is placed in a room with a key of color A and two doors of color A and B. The door of color A opens onto a room with a key of color B. The door of color B opens onto a room with a ball.
 - Instruction: Pick up the ball
-- Level id: `BabyAI-UnlockToUnlock-v0`
 - Evaluate: image understanding
+- Level id: `BabyAI-UnlockToUnlock-v0`
 
 <p align="center"><img src="media/UnlockToUnlock.png" width="250"></p>
 
 ### KeyInBox
 
 - Environment: The agent is placed in a room with a box containing a key and a locked door.
 - Instruction: Open the door
-- Level id: `BabyAI-KeyInBox-v0`
 - Evaluate: image understanding
+- Level id: `BabyAI-KeyInBox-v0`
 
 <p align="center"><img src="media/KeyInBox.png" width="250"></p>
 
 ### PickupDist
 
 - Environment: The agent is placed in a room with 5 objects. The environment is done when the instruction is executed in the regular mode or when any object is picked in the `debug` mode.
 - Instruction: Pick up an object of a given type and color
-- Level id: `BabyAI-PickupDist-v0`, `BabyAI-PickupDistDebug-v0`
 - Evaluate: image & text understanding
+- Level id:
+    - `BabyAI-PickupDist-v0`
+    - `BabyAI-PickupDistDebug-v0`
 
 <p align="center"><img src="media/PickupDist.png" width="250"></p>
 
 ### PickupAbove
 
 - Environment: The agent is placed in the middle room. An object is placed in the top-middle room.
 - Instruction: Pick up an object of a given type and color
-- Level id: `BabyAI-PickupAbove-v0`
 - Evaluate: image & text understanding, memory
+- Level id: `BabyAI-PickupAbove-v0`
 
 <p align="center"><img src="media/PickupAbove.png" width="250"></p>
 
 ### OpenRedBlueDoors
 
 - Environment: The agent is placed in a room with a red door and a blue door facing each other. The environment is done when the instruction is executed in the regular mode or when the blue door is opened in the `debug` mode.
 - Instruction: Open the red door then open the blue door
-- Level id: `BabyAI-OpenRedBlueDoors-v0`, `BabyAI-OpenRedBlueDoorsDebug-v0`
 - Evaluate: image understanding, memory
+- Level id:
+    - `BabyAI-OpenRedBlueDoors-v0`
+    - `BabyAI-OpenRedBlueDoorsDebug-v0`
 
 <p align="center"><img src="media/OpenRedBlueDoors.png" width="250"></p>
 
 ### OpenTwoDoors
 
 - Environment: The agent is placed in a room with a red door and a blue door facing each other. The environment is done when the instruction is executed in the regular mode or when the second door is opened in the `debug` mode.
 - Instruction: Open the door of color X then open the door of color Y
-- Level id: `BabyAI-OpenTwoDoors-v0`, `BabyAI-OpenTwoDoorsDebug-v0`
 - Evaluate: image & text understanding, memory
+- Level id:
+    - `BabyAI-OpenTwoDoors-v0`
+    - `BabyAI-OpenTwoDoorsDebug-v0`
 
 <p align="center"><img src="media/OpenTwoDoors.png" width="250"></p>
 
 ### FindObj
 
 - Environment: The agent is placed in the middle room. An object is placed in one of the rooms. Rooms have a size of 5 in `FindObjS5`, 6 in `FindObjS6` or 7 in `FindObjS7`.
 - Instruction: Pick up an object of a given type and color
-- Level id: `BabyAI-FindObjS5-v0`, `BabyAI-FindObjS6-v0`, `BabyAI-FindObjS7-v0`
 - Evaluate: image understanding, memory
+- Level id:
+    - `BabyAI-FindObjS5-v0`
+    - `BabyAI-FindObjS6-v0`
+    - `BabyAI-FindObjS7-v0`
 
 <p align="center">
     <img src="media/FindObjS5.png" width="250">
@@ -267,8 +284,11 @@ A work-in-progress review of related work can be found [here](https://www.overle
 
 - Environment: The agent is placed in the middle room. 4 different objects are placed in the adjacent rooms. Rooms have a size of 5 in `FourObjsS5`, 6 in `FourObjsS6` or 7 in `FourObjsS7`.
 - Instruction: Pick up an object of a given type and location
-- Level id: `BabyAI-FourObjsS5-v0`, `BabyAI-FourObjsS6-v0`, `BabyAI-FourObjsS7-v0`
 - Evaluate: image understanding, memory
+- Level id:
+    - `BabyAI-FourObjsS5-v0`
+    - `BabyAI-FourObjsS6-v0`
+    - `BabyAI-FourObjsS7-v0`
 
 <p align="center">
     <img src="media/FourObjsS5.png" width="250">
@@ -289,7 +309,14 @@ A work-in-progress review of related work can be found [here](https://www.overle
   - `BabyAI-KeyCorridorS5R3-v0`
   - `BabyAI-KeyCorridorS6R3-v0`
 
-<p align="center"><img src="media/HiddenKeyCorridor.png" width="250"></p>
+<p align="center">
+    <img src="media/KeyCorridorS3R1.png" width="250">
+    <img src="media/KeyCorridorS3R2.png" width="250">
+    <img src="media/KeyCorridorS3R3.png" width="250">
+    <img src="media/KeyCorridorS4R3.png" width="250">
+    <img src="media/KeyCorridorS5R3.png" width="250">
+    <img src="media/KeyCorridorS6R3.png" width="250">
+</p>
 
 ### 1Room
 
@@ -307,4 +334,4 @@ A work-in-progress review of related work can be found [here](https://www.overle
     <img src="media/1RoomS12.png" width="250">
     <img src="media/1RoomS16.png" width="250">
     <img src="media/1RoomS20.png" width="250">
-</p>
+</p>

babyai/levels/levels.py

@@ -698,7 +698,7 @@ def __init__(self, seed=None):
             seed=seed
         )
 
-class HiddenKeyCorridor(RoomGridLevel):
+class KeyCorridor(RoomGridLevel):
     """
     A ball is behind a locked door, the key is placed in a
     random room.
@@ -743,47 +743,47 @@ def gen_mission(self):
 
         self.instrs = [Instr(action="pickup", object=Object(obj.type))]
 
-class Level_KeyCorridorS3R1(HiddenKeyCorridor):
+class Level_KeyCorridorS3R1(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=3,
             num_rows=1,
             seed=seed
         )
 
-class Level_KeyCorridorS3R2(HiddenKeyCorridor):
+class Level_KeyCorridorS3R2(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=3,
             num_rows=2,
             seed=seed
         )
 
-class Level_KeyCorridorS3R3(HiddenKeyCorridor):
+class Level_KeyCorridorS3R3(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=3,
             num_rows=3,
             seed=seed
         )
 
-class Level_KeyCorridorS4R3(HiddenKeyCorridor):
+class Level_KeyCorridorS4R3(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=4,
             num_rows=3,
             seed=seed
         )
 
-class Level_KeyCorridorS5R3(HiddenKeyCorridor):
+class Level_KeyCorridorS5R3(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=5,
             num_rows=3,
             seed=seed
         )
 
-class Level_KeyCorridorS6R3(HiddenKeyCorridor):
+class Level_KeyCorridorS6R3(KeyCorridor):
     def __init__(self, seed=None):
         super().__init__(
             room_size=6,