Skip to content

Commit ef2b05a

Browse files
[RLlib; Offline RL] Add Offline Evaluation to Offline RL API. (#52308)
1 parent 5fc7a0e commit ef2b05a

17 files changed

+2728
-46
lines changed

rllib/BUILD

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -562,6 +562,29 @@ py_test(
562562
],
563563
)
564564

565+
py_test(
566+
name = "learning_tests_cartpole_bc_with_offline_evaluation",
567+
size = "medium",
568+
srcs = ["tuned_examples/bc/cartpole_bc_with_offline_evaluation.py"],
569+
args = [
570+
"--as-test",
571+
"--enable-new-api-stack",
572+
],
573+
# Include the offline data files.
574+
data = [
575+
"tests/data/cartpole/cartpole-v1_large",
576+
],
577+
main = "tuned_examples/bc/cartpole_bc_with_offline_evaluation.py",
578+
tags = [
579+
"exclusive",
580+
"learning_tests",
581+
"learning_tests_discrete",
582+
"learning_tests_pytorch_use_all_core",
583+
"team:rllib",
584+
"torch_only",
585+
],
586+
)
587+
565588
# CQL
566589
# Pendulum
567590
py_test(
@@ -2640,6 +2663,34 @@ py_test(
26402663
],
26412664
)
26422665

2666+
py_test(
2667+
name = "test_offline_evaluation_runner",
2668+
size = "medium",
2669+
srcs = ["offline/tests/test_offline_evaluation_runner.py"],
2670+
# Include the offline data files.
2671+
data = [
2672+
"tests/data/cartpole/cartpole-v1_large",
2673+
],
2674+
tags = [
2675+
"offline",
2676+
"team:rllib",
2677+
],
2678+
)
2679+
2680+
py_test(
2681+
name = "test_offline_evaluation_runner_group",
2682+
size = "medium",
2683+
srcs = ["offline/tests/test_offline_evaluation_runner_group.py"],
2684+
# Include the offline data files.
2685+
data = [
2686+
"tests/data/cartpole/cartpole-v1_large",
2687+
],
2688+
tags = [
2689+
"offline",
2690+
"team:rllib",
2691+
],
2692+
)
2693+
26432694
# TODO (sven, simon): This runs fine locally, but fails in the CI
26442695
# py_test(
26452696
# # TODO(#50340): test is flaky.

0 commit comments

Comments
 (0)