This allows wakelock releases to be state based instead of transition based. #exo-offload PiperOrigin-RevId: 337286501