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, [], []