From b20b70db48caa8995df5c90596f7ff73e5eb4a72 Mon Sep 17 00:00:00 2001 From: Noah Dormann Date: Thu, 21 Nov 2019 11:51:47 +0100 Subject: [PATCH] Clean reformat --- tests/test_save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_save_load.py b/tests/test_save_load.py index aff6481..01b0a25 100644 --- a/tests/test_save_load.py +++ b/tests/test_save_load.py @@ -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