fix(linting + format)
This commit is contained in:
parent
8b18a9f550
commit
968394faa8
2 changed files with 1 additions and 4 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Reference in a new issue