saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-14 20:58:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix tests and warnings when running locally with a GPU (#2069)

Browse source 
* Fix test when GPU is available

* Sort file list for consistent results

* Ignore A2C warnings too
This commit is contained in:
Antonin RAFFIN 2025-01-07 14:19:05 +01:00 committed by Antonin Raffin
parent 593b2d967f
commit 9555dd9859
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_save_load.py
View file

@ -758,16 +758,16 @@ def test_no_resource_warning(tmp_path):
# check that files are properly closed
# Create a PPO agent and save it
PPO("MlpPolicy", "CartPole-v1").save(tmp_path / "dqn_cartpole")
PPO.load(tmp_path / "dqn_cartpole")
PPO("MlpPolicy", "CartPole-v1", device="cpu").save(tmp_path / "dqn_cartpole")
PPO.load(tmp_path / "dqn_cartpole", device="cpu")
PPO("MlpPolicy", "CartPole-v1").save(str(tmp_path / "dqn_cartpole"))
PPO.load(str(tmp_path / "dqn_cartpole"))
PPO("MlpPolicy", "CartPole-v1", device="cpu").save(str(tmp_path / "dqn_cartpole"))
PPO.load(str(tmp_path / "dqn_cartpole"), device="cpu")
# Do the same but in memory, should not close the file
with tempfile.TemporaryFile() as fp:
PPO("MlpPolicy", "CartPole-v1").save(fp)
PPO.load(fp)
PPO("MlpPolicy", "CartPole-v1", device="cpu").save(fp)
PPO.load(fp, device="cpu")
assert not fp.closed
# Same but with replay buffer