From 02519b7a6e6c8fdc446bd1d76143ce0992d9d6da Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kacper=20R=C4=85czy?= Date: Fri, 14 Mar 2025 18:24:41 -0400 Subject: [PATCH] process_replay: refactor migrate_deviceState (#34864) * Refactor migrate_deviceState * Fix again --- selfdrive/test/process_replay/migration.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/selfdrive/test/process_replay/migration.py b/selfdrive/test/process_replay/migration.py index 2d42d31619..fd13c4f239 100644 --- a/selfdrive/test/process_replay/migration.py +++ b/selfdrive/test/process_replay/migration.py @@ -241,14 +241,16 @@ def migrate_gpsLocation(msgs): @migration(inputs=["deviceState", "initData"]) def migrate_deviceState(msgs): + init_data = next((m.initData for _, m in msgs if m.which() == 'initData'), None) + device_state = next((m.deviceState for _, m in msgs if m.which() == 'deviceState'), None) + if init_data is None or device_state is None: + return [], [], [] + ops = [] - dt = None for i, msg in msgs: - if msg.which() == 'initData': - dt = msg.initData.deviceType if msg.which() == 'deviceState': n = msg.as_builder() - n.deviceState.deviceType = dt + n.deviceState.deviceType = init_data.deviceType ops.append((i, n.as_reader())) return ops, [], []