-
Notifications
You must be signed in to change notification settings - Fork 56
/
__init__.py
114 lines (95 loc) · 2.79 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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
""" Define all environments and provide helper functions to load environments. """
# OpenAI gym interface
from gym.envs.registration import register
from .base import make_env, make_vec_env
# register all environment to use
from .furniture_baxter import FurnitureBaxterEnv
from .furniture_cursor import FurnitureCursorEnv
from .furniture_jaco import FurnitureJacoEnv
from .furniture_panda import FurniturePandaEnv
from .furniture_sawyer import FurnitureSawyerEnv
from .furniture_fetch import FurnitureFetchEnv
from .furniture_sawyer_dense import FurnitureSawyerDenseRewardEnv
# add cursor environment to Gym
register(
id="IKEACursor-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEACursor-v0",
"name": "FurnitureCursorEnv",
"furniture_id": 0,
"background": "Lab",
"port": 1050,
},
)
# add sawyer environment to Gym
register(
id="IKEASawyer-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEASawyer-v0",
"name": "FurnitureSawyerEnv",
"furniture_name": "swivel_chair_0700",
"background": "Industrial",
"port": 1050,
},
)
# add baxter environment to Gym
register(
id="IKEABaxter-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEABaxter-v0",
"name": "FurnitureBaxterEnv",
"furniture_id": 1,
"background": "Interior",
"port": 1050,
},
)
# add jaco environment to Gym
register(
id="IKEAJaco-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEAJaco-v0",
"name": "FurnitureJacoEnv",
"furniture_id": 1,
"background": "Interior",
"port": 1050,
},
)
# add panda environment to Gym
register(
id="IKEAPanda-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEAPanda-v0",
"name": "FurniturePandaEnv",
"furniture_id": 1,
"background": "Interior",
"port": 1050,
},
)
# add panda environment to Gym
register(
id="IKEAFetch-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={
"id": "IKEAFetch-v0",
"name": "FurnitureFetchEnv",
"furniture_id": 59,
"background": "Interior",
"port": 1050,
},
)
# add sawyer dense reward environment to Gym
register(
id="IKEASawyerDense-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={"id": "IKEASawyerDense-v0", "name": "FurnitureSawyerDenseRewardEnv", "unity": False},
)
register(
id="furniture-sawyer-densereward-v0",
entry_point="furniture.env.furniture_gym:FurnitureGym",
kwargs={"id": "IKEASawyerDense-v0", "name": "FurnitureSawyerDenseRewardEnv", "unity": False},
)