Remove redundant get

DLR-RM · May 25, 2024 · 2d62322 · 2d62322
1 parent cee4d38
commit 2d62322
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]