From 12e58ab29983f5109d0121384c646992964ec3f8 Mon Sep 17 00:00:00 2001 From: Sven Mika Date: Tue, 5 Nov 2024 18:19:43 +0100 Subject: [PATCH] [RLlib] Forward fix broken release test due to old config args being used. (#48563) Forward fix broken release test due to old config args being used. Signed-off-by: sven1977 --- release/ml_user_tests/tune_rllib/run_connect_tests.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/release/ml_user_tests/tune_rllib/run_connect_tests.py b/release/ml_user_tests/tune_rllib/run_connect_tests.py index 7fb4b2e73ccb..8c9230044adb 100644 --- a/release/ml_user_tests/tune_rllib/run_connect_tests.py +++ b/release/ml_user_tests/tune_rllib/run_connect_tests.py @@ -26,16 +26,20 @@ def run(smoke_test=False, storage_path: str = None): config = ( APPOConfig() + .api_stack( + enable_rl_module_and_learner=False, + enable_env_runner_and_connector_v2=False, + ) .environment("ale_py:ALE/Pong-v5", clip_rewards=True) .framework(tune.grid_search(["tf", "torch"])) .rollouts( rollout_fragment_length=50, - num_rollout_workers=num_workers, - num_envs_per_worker=1, + num_env_runners=num_workers, + num_envs_per_env_runner=1, ) .training( train_batch_size=750, - num_sgd_iter=2, + num_epochs=2, vf_loss_coeff=1.0, clip_param=0.3, grad_clip=10,