From b3dc7bb3cb75661993c068b993712a3a9f5238a2 Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Tue, 16 Aug 2022 18:40:30 -0700 Subject: [PATCH] regen: initialize once (#25458) regen: init once --- selfdrive/test/process_replay/regen.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/selfdrive/test/process_replay/regen.py b/selfdrive/test/process_replay/regen.py index 39f75ce428..16d7ea8519 100755 --- a/selfdrive/test/process_replay/regen.py +++ b/selfdrive/test/process_replay/regen.py @@ -201,14 +201,14 @@ def regen_segment(lr, frs=None, outdir=FAKEDATA, disable_tqdm=False): params = Params() os.environ["LOG_ROOT"] = outdir - for msg in lr: - if msg.which() == 'carParams': - setup_env(CP=msg.carParams) - elif msg.which() == 'liveCalibration': - params.put("CalibrationParams", msg.as_builder().to_bytes()) + # Get and setup initial state + CP = [m for m in lr if m.which() == 'carParams'][0].carParams + liveCalibration = [m for m in lr if m.which() == 'liveCalibration'][0] - vs, cam_procs = replay_cameras(lr, frs, disable_tqdm=disable_tqdm) + setup_env(CP=CP) + params.put("CalibrationParams", liveCalibration.as_builder().to_bytes()) + vs, cam_procs = replay_cameras(lr, frs, disable_tqdm=disable_tqdm) fake_daemons = { 'sensord': [ multiprocessing.Process(target=replay_sensor_events, args=('sensorEvents', lr)),