From e1fe30fd3e5b3ac03cc4503022191e383c873400 Mon Sep 17 00:00:00 2001 From: James Vecellio-Grant <159560811+Discountchubbs@users.noreply.github.com> Date: Sun, 7 Jun 2026 12:19:36 +0200 Subject: [PATCH] Update compile_modeld.py --- sunnypilot/modeld_v2/compile_modeld.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sunnypilot/modeld_v2/compile_modeld.py b/sunnypilot/modeld_v2/compile_modeld.py index b9a392631a..4b4d0c2c3a 100755 --- a/sunnypilot/modeld_v2/compile_modeld.py +++ b/sunnypilot/modeld_v2/compile_modeld.py @@ -240,9 +240,9 @@ if __name__ == "__main__": assert vision_runner policy_runners, policy_names = _load_policy_runners(args) output_data['metadata'] = {'vision': make_metadata_dict(args.vision_onnx)} - for name, runner_arg in zip(policy_names, [args.policy_onnx, args.off_policy_onnx, args.on_policy_onnx], strict=True): - if runner_arg: - output_data['metadata'][name] = make_metadata_dict(runner_arg) + for name in policy_names: + runner_arg = getattr(args, f"{name}_onnx") + output_data['metadata'][name] = make_metadata_dict(runner_arg) first_policy_meta = output_data['metadata'].get('policy', output_data['metadata'].get('model', output_data['metadata'].get('off_policy', {}))) vision_meta = output_data['metadata'].get('vision', {})