diff --git a/cereal/libcereal.a b/cereal/libcereal.a index 89c2da8ca..9e78a24b8 100644 Binary files a/cereal/libcereal.a and b/cereal/libcereal.a differ diff --git a/cereal/log.capnp b/cereal/log.capnp index 751f8fa99..1212e1982 100644 --- a/cereal/log.capnp +++ b/cereal/log.capnp @@ -2192,6 +2192,7 @@ struct DriverStateV2 { rightBlinkProb @8 :Float32; sunglassesProb @9 :Float32; phoneProb @13 :Float32; + sleepProb @14 :Float32; notReadyProbDEPRECATED @12 :List(Float32); occludedProbDEPRECATED @10 :Float32; readyProbDEPRECATED @11 :List(Float32); diff --git a/selfdrive/modeld/dmonitoringmodeld.py b/selfdrive/modeld/dmonitoringmodeld.py index 1b63009b5..383b52571 100644 --- a/selfdrive/modeld/dmonitoringmodeld.py +++ b/selfdrive/modeld/dmonitoringmodeld.py @@ -114,10 +114,7 @@ def fill_driver_data(msg, model_output, suffix): msg.rightBlinkProb = model_output[f"right_blink_prob_{suffix}"][0, 0].item() msg.sunglassesProb = model_output[f"sunglasses_prob_{suffix}"][0, 0].item() msg.phoneProb = model_output[f"using_phone_prob_{suffix}"][0, 0].item() - try: - msg.sleepProb = model_output[f"sleep_prob_{suffix}"][0, 0].item() - except AttributeError: - pass + msg.sleepProb = model_output[f"sleep_prob_{suffix}"][0, 0].item() def get_driverstate_packet(model_output, frame_id: int, exec_time: float, gpu_exec_time: float):