Skip to content

Commit

Permalink
[RLlib] Require ApeX LR schedule test to produce learner info. (ray-p…
Browse files Browse the repository at this point in the history
…roject#27557)

Signed-off-by: Stefan van der Kleij <[email protected]>
  • Loading branch information
ArturNiederfahrenhorst authored and Stefan van der Kleij committed Aug 18, 2022
1 parent d2cfb54 commit 2431ac3
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions rllib/algorithms/apex_dqn/tests/test_apex_dqn.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,6 +124,8 @@ def test_apex_lr_schedule(self):
)
.reporting(
min_sample_timesteps_per_iteration=10,
# Make sure that results contain info on default policy
min_train_timesteps_per_iteration=10,
# 0 metrics reporting delay, this makes sure timestep,
# which lr depends on, is updated after each worker rollout.
min_time_s_per_iteration=0,
Expand Down

0 comments on commit 2431ac3

Please sign in to comment.