diff --git a/rllib/BUILD b/rllib/BUILD index 74fcb336b450..7c2259b84768 100644 --- a/rllib/BUILD +++ b/rllib/BUILD @@ -2836,107 +2836,6 @@ py_test( args = ["--enable-new-api-stack", "--as-test"] ) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_with_torch_lr_schedulers", - main = "examples/learners/ppo_with_torch_lr_schedulers.py", - tags = ["team:rllib", "examples"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=torch", "--config=local-cpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_local_cpu_tf2", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=tf2", "--config=local-cpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_local_gpu_torch", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "gpu"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=torch", "--config=local-gpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_local_gpu_tf2", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "gpu", "exclusive"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=tf2", "--config=local-gpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_remote_cpu_torch", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=torch", "--config=remote-cpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_remote_cpu_tf2", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=tf2", "--config=remote-cpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_remote_gpu_torch", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "gpu", "exclusive"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=torch", "--config=remote-gpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_remote_gpu_tf2", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "gpu", "exclusive"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=tf2", "--config=remote-gpu"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_multi_gpu_torch", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "multi_gpu", "exclusive"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=torch", "--config=multi-gpu-ddp"] -) - -#@OldAPIStack @HybridAPIStack -py_test( - name = "examples/learners/ppo_tuner_multi_gpu_tf2", - main = "examples/learners/ppo_tuner.py", - tags = ["team:rllib", "examples", "multi_gpu", "exclusive"], - size = "medium", - srcs = ["examples/learners/ppo_tuner.py"], - args = ["--framework=tf2", "--config=multi-gpu-ddp"] -) - # subdirectory: multi_agent/ # .................................... py_test( @@ -3256,56 +3155,6 @@ py_test( args = ["--as-test", "--framework=torch", "--stop-reward=-0.012", "--num-cpus=4"] ) -#@OldAPIStack -py_test( - name = "examples/cartpole_lstm_impala_tf2", - main = "examples/cartpole_lstm.py", - tags = ["team:rllib", "exclusive", "examples"], - size = "medium", - srcs = ["examples/cartpole_lstm.py"], - args = ["--run=IMPALA", "--as-test", "--framework=tf2", "--stop-reward=28", "--num-cpus=4"] -) - -#@OldAPIStack -py_test( - name = "examples/cartpole_lstm_impala_torch", - main = "examples/cartpole_lstm.py", - tags = ["team:rllib", "exclusive", "examples"], - size = "medium", - srcs = ["examples/cartpole_lstm.py"], - args = ["--run=IMPALA", "--as-test", "--framework=torch", "--stop-reward=28", "--num-cpus=4"] -) - -#@OldAPIStack -py_test( - name = "examples/cartpole_lstm_ppo_tf2", - main = "examples/cartpole_lstm.py", - tags = ["team:rllib", "exclusive", "examples"], - size = "large", - srcs = ["examples/cartpole_lstm.py"], - args = ["--run=PPO", "--as-test", "--framework=tf2", "--stop-reward=28", "--num-cpus=4"] -) - -#@OldAPIStack -py_test( - name = "examples/cartpole_lstm_ppo_torch", - main = "examples/cartpole_lstm.py", - tags = ["team:rllib", "exclusive", "examples"], - size = "medium", - srcs = ["examples/cartpole_lstm.py"], - args = ["--run=PPO", "--as-test", "--framework=torch", "--stop-reward=28", "--num-cpus=4"] -) - -#@OldAPIStack -py_test( - name = "examples/cartpole_lstm_ppo_torch_with_prev_a_and_r", - main = "examples/cartpole_lstm.py", - tags = ["team:rllib", "exclusive", "examples"], - size = "medium", - srcs = ["examples/cartpole_lstm.py"], - args = ["--run=PPO", "--as-test", "--framework=torch", "--stop-reward=28", "--num-cpus=4", "--use-prev-action", "--use-prev-reward"] -) - #@OldAPIStack py_test( name = "examples/centralized_critic_tf",