From 4fe47d069f0a3af663648c5075f8b2b0881568e2 Mon Sep 17 00:00:00 2001 From: Artur Niederfahrenhorst Date: Mon, 8 Aug 2022 18:19:02 +0200 Subject: [PATCH] [RLlib] Require ApeX LR schedule test to produce learner info. (#27557) --- rllib/algorithms/apex_dqn/tests/test_apex_dqn.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rllib/algorithms/apex_dqn/tests/test_apex_dqn.py b/rllib/algorithms/apex_dqn/tests/test_apex_dqn.py index 7d97a4ddf..9620f0ec6 100644 --- a/rllib/algorithms/apex_dqn/tests/test_apex_dqn.py +++ b/rllib/algorithms/apex_dqn/tests/test_apex_dqn.py @@ -124,6 +124,8 @@ class TestApexDQN(unittest.TestCase): ) .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,