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

fix: universal .flax files

Browse source
This commit is contained in:
Jona Reynaert 2026-04-16 10:48:37 +02:00
parent f6022cc913
commit cb6e3d9714
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
scripts/simulate.py
View file

@ -373,9 +373,9 @@ def main() -> None:
nu = int(state.mj_model.nu)
model_path = Path(args.model)
if model_path.name != "final_model.flax" or model_path.suffix != ".flax":
if model_path.suffix != ".flax":
raise ValueError(
f"Expected the training artifact 'final_model.flax', got '{model_path.name}'."
f"Expected the training artifact '.flax', got '{model_path.name}'."
)
policy = CleanRLPPOPolicy.load(