diff --git a/selfdrive/monitoring/test_monitoring.py b/selfdrive/monitoring/test_monitoring.py index a84ed242ba..1d903a2519 100755 --- a/selfdrive/monitoring/test_monitoring.py +++ b/selfdrive/monitoring/test_monitoring.py @@ -18,18 +18,18 @@ INVISIBLE_SECONDS_TO_RED = dm_settings._AWARENESS_TIME + 1 def make_msg(face_detected, distracted=False, model_uncertain=False): ds = log.DriverState.new_message() - ds.faceOrientation = [0., 0., 0.] - ds.facePosition = [0., 0.] - ds.faceProb = 1. * face_detected - ds.leftEyeProb = 1. - ds.rightEyeProb = 1. - ds.leftBlinkProb = 1. * distracted - ds.rightBlinkProb = 1. * distracted - ds.faceOrientationStd = [1.*model_uncertain, 1.*model_uncertain, 1.*model_uncertain] - ds.facePositionStd = [1.*model_uncertain, 1.*model_uncertain] + ds.driverDataLH.faceOrientation = [0., 0., 0.] + ds.driverDataLH.facePosition = [0., 0.] + ds.driverDataLH.faceProb = 1. * face_detected + ds.driverDataLH.leftEyeProb = 1. + ds.driverDataLH.rightEyeProb = 1. + ds.driverDataLH.leftBlinkProb = 1. * distracted + ds.driverDataLH.rightBlinkProb = 1. * distracted + ds.driverDataLH.faceOrientationStd = [1.*model_uncertain, 1.*model_uncertain, 1.*model_uncertain] + ds.driverDataLH.facePositionStd = [1.*model_uncertain, 1.*model_uncertain] # TODO: test both separately when e2e is used - ds.readyProb = [0., 0., 0., 0.] - ds.notReadyProb = [0., 0.] + ds.driverDataLH.readyProb = [0., 0., 0., 0.] + ds.driverDataLH.notReadyProb = [0., 0.] return ds