1
Fork 0

fix(linting + format)

This commit is contained in:
Robin Meersman 2026-05-11 21:14:44 +02:00
parent 8b18a9f550
commit 968394faa8
2 changed files with 1 additions and 4 deletions

View file

@ -135,9 +135,7 @@ def main(dict_cfg: DictConfig) -> None:
trained_action_dim = raw_env.action_space.shape[0] // needed_copies
# 7. Load policy
message_passing_steps = (
metadata.get("architecture", {}) or {}
).get("message_passing_steps")
message_passing_steps = (metadata.get("architecture", {}) or {}).get("message_passing_steps")
if message_passing_steps is None:
message_passing_steps = 4
message_passing_steps = int(message_passing_steps)