From 6daf82bf7439675fa8595812ddaa1bc00473da26 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Sun, 31 Oct 2021 19:03:28 +0100 Subject: [PATCH] Relax test --- tests/test_logger.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_logger.py b/tests/test_logger.py index 57463b0..6cb7385 100644 --- a/tests/test_logger.py +++ b/tests/test_logger.py @@ -337,11 +337,11 @@ def test_fps_logger(tmp_path, algo): # fps should be at most max_fps model.learn(100, log_interval=1) - assert max_fps / 4 <= logger.name_to_value["time/fps"] <= max_fps + assert max_fps / 10 <= logger.name_to_value["time/fps"] <= max_fps # second time, FPS should be the same model.learn(100, log_interval=1) - assert max_fps / 4 <= logger.name_to_value["time/fps"] <= max_fps + assert max_fps / 10 <= logger.name_to_value["time/fps"] <= max_fps # Artificially increase num_timesteps to check # that fps computation is reset at each call to learn() @@ -349,4 +349,4 @@ def test_fps_logger(tmp_path, algo): # third time, FPS should be the same model.learn(100, log_interval=1, reset_num_timesteps=False) - assert max_fps / 4 <= logger.name_to_value["time/fps"] <= max_fps + assert max_fps / 10 <= logger.name_to_value["time/fps"] <= max_fps