[RLlib] Forward fix broken release test due to old config args being …

…used. (ray-project#48563) Forward fix broken release test due to old config args being used. Signed-off-by: sven1977 <[email protected]> Signed-off-by: mohitjain2504 <[email protected]>
mohitjain2504 · Nov 15, 2024 · 755cacd · 755cacd
1 parent f01d4ae
commit 755cacd
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,