[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 <svenmika1977@gmail.com>
ray-project · Nov 5, 2024 · 12e58ab · 12e58ab
1 parent 7aa3cdb
commit 12e58ab
Showing 1 changed file with 7 additions and 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
@@ -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,