From 2d62322f26048113e3a47b51af1cb732e132942d Mon Sep 17 00:00:00 2001 From: Joe Ksiazek Date: Sat, 25 May 2024 18:36:57 -0400 Subject: [PATCH] Remove redundant get --- stable_baselines3/common/base_class.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_baselines3/common/base_class.py b/stable_baselines3/common/base_class.py index f7d4d97bb..b2c967405 100644 --- a/stable_baselines3/common/base_class.py +++ b/stable_baselines3/common/base_class.py @@ -692,7 +692,7 @@ def load( # noqa: C901 if "device" in data["policy_kwargs"]: del data["policy_kwargs"]["device"] # backward compatibility, convert to new format - saved_net_arch = data.get("policy_kwargs", {}).get("net_arch") + saved_net_arch = data["policy_kwargs"].get("net_arch") if saved_net_arch and isinstance(saved_net_arch, list) and isinstance(saved_net_arch[0], dict): data["policy_kwargs"]["net_arch"] = saved_net_arch[0]