From 15e1415d3cbe58bf14814d487b149b9dcdc0c1e6 Mon Sep 17 00:00:00 2001 From: Shane Smiskol Date: Thu, 3 Oct 2024 13:08:10 -0700 Subject: [PATCH] migrate dm events --- selfdrive/test/process_replay/migration.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/selfdrive/test/process_replay/migration.py b/selfdrive/test/process_replay/migration.py index 80f397cc15..920fce938f 100644 --- a/selfdrive/test/process_replay/migration.py +++ b/selfdrive/test/process_replay/migration.py @@ -38,6 +38,7 @@ def migrate_all(lr: LogIterable, manager_states: bool = False, panda_states: boo migrate_driverAssistance, migrate_drivingModelData, migrate_onroadEvents, + migrate_driverMonitoringState, ] if manager_states: migrations.append(migrate_managerState) @@ -408,3 +409,16 @@ def migrate_onroadEvents(msgs): ops.append((index, new_msg.as_reader())) return ops, [], [] + + +@migration(inputs=["driverMonitoringState"]) +def migrate_driverMonitoringState(msgs): + ops = [] + for index, msg in msgs: + msg = msg.as_builder() + # dict converts name enum into string representation + msg.driverMonitoringState.events = [log.OnroadEvent(**event.to_dict()) for event in + msg.driverMonitoringState.eventsDEPRECATED] + ops.append((index, msg.as_reader())) + + return ops, [], []