@@ -562,6 +562,29 @@ py_test(
562
562
],
563
563
)
564
564
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
+
565
588
# CQL
566
589
# Pendulum
567
590
py_test (
@@ -2640,6 +2663,34 @@ py_test(
2640
2663
],
2641
2664
)
2642
2665
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
+
2643
2694
# TODO (sven, simon): This runs fine locally, but fails in the CI
2644
2695
# py_test(
2645
2696
# # TODO(#50340): test is flaky.
0 commit comments