Skip to content

Commit 7780832

Browse files
authored
Fix key check in on_policy_runner.py (#60)
* Update on_policy_runner.py
1 parent 1cba8fb commit 7780832

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: rsl_rl/runners/on_policy_runner.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ def __init__(self, env: VecEnv, train_cfg: dict, log_dir: str | None = None, dev
4141
).to(self.device)
4242

4343
# resolve dimension of rnd gated state
44-
if self.alg_cfg["rnd_cfg"] is not None:
44+
if "rnd_cfg" in self.alg_cfg:
4545
# check if rnd gated state is present
4646
rnd_state = extras["observations"].get("rnd_state")
4747
if rnd_state is None:
@@ -54,7 +54,7 @@ def __init__(self, env: VecEnv, train_cfg: dict, log_dir: str | None = None, dev
5454
self.alg_cfg["rnd_cfg"]["weight"] *= env.dt
5555

5656
# if using symmetry then pass the environment config object
57-
if self.alg_cfg["symmetry_cfg"] is not None:
57+
if "symmetry_cfg" in self.alg_cfg:
5858
# this is used by the symmetry function for handling different observation terms
5959
self.alg_cfg["symmetry_cfg"]["_env"] = env
6060

0 commit comments

Comments
 (0)