tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
tdpeuter/2026SEL3-project-Brittle_Star_Controller_Modularity
Archived
1
Fork 0
Code Issues Releases Activity

fix(linting + format)

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
scripts/simulate.py
View file

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

1
scripts/train.py
View file

@ -8,7 +8,6 @@ from brittle_star_project.configs.register_configs import register_configs
from brittle_star_project.trainers.PPOTrainer import PPOTrainer from brittle_star_project.trainers.PPOTrainer import PPOTrainer
from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper from brittle_star_project.environment.BrittleStarJaxEnvWrapper import BrittleStarJaxEnvWrapper
from experiment_logger import init_logger, get_logger from experiment_logger import init_logger, get_logger
import logging
def make_env(cfg: BrittleStarConfig) -> BrittleStarJaxEnvWrapper: def make_env(cfg: BrittleStarConfig) -> BrittleStarJaxEnvWrapper: