Skip to content

Commit 12644f1

Browse files
committed
Replace VizDoom pretrain test with CV policy
1 parent 9b7653e commit 12644f1

2 files changed

Lines changed: 588 additions & 216 deletions

File tree

envpool/vizdoom/BUILD

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -19,18 +19,6 @@ load("//envpool:requirements.bzl", "requirement")
1919

2020
package(default_visibility = ["//visibility:public"])
2121

22-
genrule(
23-
name = "gen_pretrain_weight",
24-
srcs = [
25-
"@pretrain_weight//:pth",
26-
],
27-
outs = [
28-
"policy-d1.pth",
29-
"policy-d3.pth",
30-
],
31-
cmd = "cp $(SRCS) $(@D)",
32-
)
33-
3422
genrule(
3523
name = "gen_vizdoom_maps",
3624
srcs = [
@@ -179,15 +167,12 @@ py_test(
179167
name = "vizdoom_pretrain_test",
180168
size = "enormous",
181169
srcs = ["vizdoom_pretrain_test.py"],
182-
data = [":gen_pretrain_weight"],
183170
imports = ["../.."],
184171
deps = [
185172
":vizdoom",
186173
":vizdoom_registration",
187-
"//envpool/atari:atari_network",
188174
requirement("numpy"),
189175
requirement("absl-py"),
190-
requirement("tianshou"),
191-
requirement("torch"),
176+
requirement("opencv-python-headless"),
192177
],
193178
)

0 commit comments

Comments
 (0)