From a7751702b7cc35a701715cdd67a7f8b01edeb690 Mon Sep 17 00:00:00 2001 From: discountchubbs Date: Tue, 8 Jul 2025 06:16:45 -0700 Subject: [PATCH] This needs to be apart of the conditional else fail --- sunnypilot/modeld_v2/parse_model_outputs_split.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sunnypilot/modeld_v2/parse_model_outputs_split.py b/sunnypilot/modeld_v2/parse_model_outputs_split.py index 9027b56b87..c0f1589b76 100644 --- a/sunnypilot/modeld_v2/parse_model_outputs_split.py +++ b/sunnypilot/modeld_v2/parse_model_outputs_split.py @@ -127,11 +127,12 @@ class Parser: return outs def parse_policy_outputs(self, outs: dict[str, np.ndarray]) -> dict[str, np.ndarray]: - self.parse_mdn('plan', outs, in_N=SplitModelConstants.PLAN_MHP_N, out_N=SplitModelConstants.PLAN_MHP_SELECTION, - out_shape=(SplitModelConstants.IDX_N,SplitModelConstants.PLAN_WIDTH)) self.split_outputs(outs) if self.zero_inputs: self.split_zero_inputs(outs) + else: + self.parse_mdn('plan', outs, in_N=SplitModelConstants.PLAN_MHP_N, out_N=SplitModelConstants.PLAN_MHP_SELECTION, + out_shape=(SplitModelConstants.IDX_N,SplitModelConstants.PLAN_WIDTH)) if 'lat_planner_solution' in outs: self.parse_mdn('lat_planner_solution', outs, in_N=0, out_N=0, out_shape=(SplitModelConstants.IDX_N,SplitModelConstants.LAT_PLANNER_SOLUTION_WIDTH)) if 'desired_curvature' in outs: