-
Notifications
You must be signed in to change notification settings - Fork 13
/
__init__.py
40 lines (37 loc) · 1.1 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
from .historicalobs import *
from .singleroom import *
from .boxkey import *
from .randomboxkey import *
from .keydistraction import *
from .Game import *
from .Game_RL import *
gym.envs.register(
id='MiniGrid-SimpleDoorKey-Min5-Max10-View3',
entry_point='env.singleroom:SingleRoomEnv',
kwargs={'minRoomSize' : 5, \
'maxRoomSize' : 10, \
'agent_view_size' : 3},
)
gym.envs.register(
id='MiniGrid-KeyInBox-Min5-Max10-View3',
entry_point='env.boxkey:BoxKeyEnv',
kwargs={'minRoomSize' : 5, \
'maxRoomSize' : 10, \
'agent_view_size' : 3},
)
gym.envs.register(
id='MiniGrid-RandomBoxKey-Min5-Max10-View3',
entry_point='env.randomboxkey:RandomBoxKeyEnv',
kwargs={'minRoomSize' : 5, \
'maxRoomSize' : 10, \
'agent_view_size' : 3},
)
gym.envs.register(
id='MiniGrid-ColoredDoorKey-Min5-Max10-View3',
entry_point='env.keydistraction:KeyDistractionEnv',
kwargs={'minRoomSize' : 5, \
'maxRoomSize' : 10, \
'minNumKeys' : 2, \
'maxNumKeys' : 2, \
'agent_view_size' : 3},
)