diff --git a/homeassistant/components/yoto/binary_sensor.py b/homeassistant/components/yoto/binary_sensor.py index d76e8e9c6629..b16fa4d2a5c4 100644 --- a/homeassistant/components/yoto/binary_sensor.py +++ b/homeassistant/components/yoto/binary_sensor.py @@ -31,6 +31,7 @@ BINARY_SENSORS: tuple[YotoBinarySensorEntityDescription, ...] = ( YotoBinarySensorEntityDescription( key="charging", device_class=BinarySensorDeviceClass.BATTERY_CHARGING, + entity_category=EntityCategory.DIAGNOSTIC, is_on_fn=lambda player: player.status.is_charging, ), YotoBinarySensorEntityDescription( diff --git a/homeassistant/components/yoto/sensor.py b/homeassistant/components/yoto/sensor.py index 0b99b2829b9e..c8d1021706b1 100644 --- a/homeassistant/components/yoto/sensor.py +++ b/homeassistant/components/yoto/sensor.py @@ -47,6 +47,7 @@ SENSORS: tuple[YotoSensorEntityDescription, ...] = ( device_class=SensorDeviceClass.BATTERY, native_unit_of_measurement=PERCENTAGE, state_class=SensorStateClass.MEASUREMENT, + entity_category=EntityCategory.DIAGNOSTIC, value_fn=lambda player: player.status.battery_level_percentage, ), YotoSensorEntityDescription( diff --git a/tests/components/yoto/snapshots/test_binary_sensor.ambr b/tests/components/yoto/snapshots/test_binary_sensor.ambr index 222c7e0f9463..15190fcbe67c 100644 --- a/tests/components/yoto/snapshots/test_binary_sensor.ambr +++ b/tests/components/yoto/snapshots/test_binary_sensor.ambr @@ -114,7 +114,7 @@ 'device_id': , 'disabled_by': None, 'domain': 'binary_sensor', - 'entity_category': None, + 'entity_category': , 'entity_id': 'binary_sensor.nursery_yoto_charging', 'has_entity_name': True, 'hidden_by': None, diff --git a/tests/components/yoto/snapshots/test_sensor.ambr b/tests/components/yoto/snapshots/test_sensor.ambr index b3da84b146f5..4fe3d542b6fc 100644 --- a/tests/components/yoto/snapshots/test_sensor.ambr +++ b/tests/components/yoto/snapshots/test_sensor.ambr @@ -14,7 +14,7 @@ 'device_id': , 'disabled_by': None, 'domain': 'sensor', - 'entity_category': None, + 'entity_category': , 'entity_id': 'sensor.nursery_yoto_battery', 'has_entity_name': True, 'hidden_by': None,