-
Notifications
You must be signed in to change notification settings - Fork 130
/
BUILD
40 lines (35 loc) · 743 Bytes
/
BUILD
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
package(default_visibility = ["//visibility:public"])
env_args = [
"--env_type lab",
"--env_name nav_maze_static_01",
"--use_pixel_change True",
"--use_value_replay True",
"--use_reward_prediction True"
]
py_binary(
name = "train",
srcs = ["main.py"],
args = env_args,
data = ["//:deepmind_lab.so"],
main = "main.py"
)
py_binary(
name = "display",
srcs = ["display.py"],
args = env_args,
data = ["//:deepmind_lab.so"],
main = "display.py"
)
py_binary(
name = "visualize",
srcs = ["visualize.py"],
args = env_args,
data = ["//:deepmind_lab.so"],
main = "visualize.py"
)
py_test(
name = "test",
srcs = ["test.py"],
main = "test.py",
deps = [":train"],
)