From cb6e3d97142fa3a8dea816835f96385774ae4874 Mon Sep 17 00:00:00 2001 From: Jona Reynaert Date: Thu, 16 Apr 2026 10:48:37 +0200 Subject: [PATCH] fix: universal .flax files --- scripts/simulate.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/simulate.py b/scripts/simulate.py index 4afdc24..38e4ae5 100644 --- a/scripts/simulate.py +++ b/scripts/simulate.py @@ -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(