Skip to content

Commit

Permalink
Fix missing test data files dependency in rllib/BUILD
Browse files Browse the repository at this point in the history
See ray-project#26334 and ray-project#26517 for context.

Once this is in, it should be good to roll-forwrad again.
  • Loading branch information
Riatre committed Jul 13, 2022
1 parent ab10890 commit 97b0ccb
Showing 1 changed file with 16 additions and 6 deletions.
22 changes: 16 additions & 6 deletions rllib/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -711,7 +711,8 @@ py_test(
name = "test_algorithm",
tags = ["team:rllib", "algorithms_dir", "algorithms_dir_generic"],
size = "large",
srcs = ["algorithms/tests/test_algorithm.py"]
srcs = ["algorithms/tests/test_algorithm.py"],
data = ["tests/data/cartpole/small.json"],
)

py_test(
Expand Down Expand Up @@ -825,6 +826,7 @@ py_test(
name = "test_cql",
tags = ["team:rllib", "algorithms_dir"],
size = "large",
data = ["tests/data/cartpole/small.json"],
srcs = ["algorithms/cql/tests/test_cql.py"]
)

Expand All @@ -833,7 +835,8 @@ py_test(
name = "test_crr",
tags = ["team:rllib", "algorithms_dir"],
size = "medium",
srcs = ["algorithms/crr/tests/test_crr.py"]
srcs = ["algorithms/crr/tests/test_crr.py"],
data = ["tests/data/pendulum/large.json"],
)

# DDPG
Expand Down Expand Up @@ -896,7 +899,11 @@ py_test(
tags = ["team:rllib", "algorithms_dir"],
size = "large",
# Include the json data file.
data = ["tests/data/cartpole/large.json"],
data = [
"tests/data/cartpole/large.json",
"tests/data/pendulum/large.json",
"tests/data/cartpole/small.json",
],
srcs = ["algorithms/marwil/tests/test_marwil.py"]
)

Expand Down Expand Up @@ -1651,7 +1658,8 @@ py_test(
name = "test_dataset_reader",
tags = ["team:rllib", "offline"],
size = "small",
srcs = ["offline/tests/test_dataset_reader.py"]
srcs = ["offline/tests/test_dataset_reader.py"],
data = ["tests/data/pendulum/large.json"],
)

py_test(
Expand All @@ -1665,14 +1673,16 @@ py_test(
name = "test_json_reader",
tags = ["team:rllib", "offline"],
size = "small",
srcs = ["offline/tests/test_json_reader.py"]
srcs = ["offline/tests/test_json_reader.py"],
data = ["tests/data/pendulum/large.json"],
)

py_test(
name = "test_ope",
tags = ["team:rllib", "offline", "torch_only"],
size = "medium",
srcs = ["offline/estimators/tests/test_ope.py"]
srcs = ["offline/estimators/tests/test_ope.py"],
data = ["tests/data/cartpole/large.json"],
)


Expand Down

0 comments on commit 97b0ccb

Please sign in to comment.