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

Clean reformat

Browse source
This commit is contained in:
Noah Dormann 2019-11-21 11:51:47 +01:00
parent 5bca52a87d
commit b20b70db48
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_save_load.py
View file

@ -50,7 +50,7 @@ def test_save_load(model_class):
model.save("test_save.zip")
model = model.load("test_save")
#check if params are still the same after load
# check if params are still the same after load
new_params = model.get_policy_parameters()
shared_items = {k: params[k] for k in params if k in new_params and th.all(th.eq(params[k], new_params[k]))}
# Check that at least some actions are chosen different now