Skip to content

Commit

Permalink
[RLlib] Forward fix broken release test due to old config args being …
Browse files Browse the repository at this point in the history
…used. (ray-project#48563)

Forward fix broken release test due to old config args being used.

Signed-off-by: sven1977 <svenmika1977@gmail.com>
Signed-off-by: mohitjain2504 <mohit.jain@dream11.com>
  • Loading branch information
sven1977 authored and mohitjain2504 committed Nov 15, 2024
1 parent f01d4ae commit 755cacd
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions release/ml_user_tests/tune_rllib/run_connect_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 755cacd

Please sign in to comment.