BUILD

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

tensorforce_args = [
  "--agent VPGAgent",
  "--agent-config /configs/vpg_baseline_visual.json",
  "--network-config /configs/cnn_dqn_network.json",
  "--episodes 1000",
  "--max-timesteps 1000"
]

py_library(
    name = "tensorforce",
    imports = [":tensorforce"],
    data = ["//tensorforce:examples/configs/vpg_baseline_visual.json",
    "//tensorforce:examples/configs/cnn_dqn_network.json"],
    srcs = glob(["tensorforce/**/*.py"])
)

py_binary(
    name = "lab_runner",
    srcs = ["examples/lab_main.py"],
    args = tensorforce_args,
    data = ["//:deepmind_lab.so"],
    main = "examples/lab_main.py",
    deps = [":tensorforce"]
)