Skip to content

Commit 9cb0ebe

Browse files
sshanepd0wm
andauthored
LongitudinalMpc: Remove class accel_limit_arr (#23295)
* remove class accel_limit_arr remove class accel_limit_arr * make it static * only set static parameters once * just use existing function style * Update selfdrive/controls/lib/longitudinal_mpc_lib/long_mpc.py * Update selfdrive/controls/lib/longitudinal_mpc_lib/long_mpc.py Co-authored-by: Willem Melching <[email protected]>
1 parent 71132ed commit 9cb0ebe

File tree

1 file changed

+4
-10
lines changed

1 file changed

+4
-10
lines changed

selfdrive/controls/lib/longitudinal_mpc_lib/long_mpc.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -194,9 +194,6 @@ class LongitudinalMpc:
194194
def __init__(self, e2e=False):
195195
self.e2e = e2e
196196
self.reset()
197-
self.accel_limit_arr = np.zeros((N+1, 2))
198-
self.accel_limit_arr[:,0] = -1.2
199-
self.accel_limit_arr[:,1] = 1.2
200197
self.source = SOURCES[2]
201198

202199
def reset(self):
@@ -224,6 +221,9 @@ def reset(self):
224221
def set_weights(self):
225222
if self.e2e:
226223
self.set_weights_for_xva_policy()
224+
self.params[:,0] = -10.
225+
self.params[:,1] = 10.
226+
self.params[:,2] = 1e5
227227
else:
228228
self.set_weights_for_lead_policy()
229229

@@ -348,15 +348,9 @@ def update_with_xva(self, x, v, a):
348348
for i in range(N):
349349
self.solver.cost_set(i, "yref", self.yref[i])
350350
self.solver.cost_set(N, "yref", self.yref[N][:COST_E_DIM])
351-
self.accel_limit_arr[:,0] = -10.
352-
self.accel_limit_arr[:,1] = 10.
353-
x_obstacle = 1e5*np.ones(N+1)
354-
self.params = np.concatenate([self.accel_limit_arr,
355-
x_obstacle[:, None],
356-
self.prev_a[:,None]], axis=1)
351+
self.params[:,3] = np.copy(self.prev_a)
357352
self.run()
358353

359-
360354
def run(self):
361355
for i in range(N+1):
362356
self.solver.set(i, 'p', self.params[i])

0 commit comments

Comments
 (0)