saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-27 03:11:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Relax test

Browse source
This commit is contained in:
Antonin Raffin 2021-10-31 19:03:28 +01:00
parent 98c1a637cf
commit 6daf82bf74
No known key found for this signature in database
GPG key ID: B8B48F65CAD6232C
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_logger.py
View file

@ -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