Correct unit_of_measurement for trigger-based template sensor (#69291)

homeassistant/components/template/sensor.py

@@ -253,6 +253,11 @@ class TriggerSensorEntity(TriggerEntity, SensorEntity):
         """Sensor state class."""
         return self._config.get(CONF_STATE_CLASS)
 
+    @property
+    def native_unit_of_measurement(self) -> str | None:
+        """Return the unit of measurement of the sensor, if any."""
+        return self._config.get(CONF_UNIT_OF_MEASUREMENT)
+
     @callback
     def _process_data(self) -> None:
         """Process new data."""

homeassistant/components/template/trigger_entity.py

@@ -4,13 +4,7 @@ from __future__ import annotations
 import logging
 from typing import Any
 
-from homeassistant.const import (
+from homeassistant.const import CONF_DEVICE_CLASS, CONF_ICON, CONF_NAME, CONF_UNIQUE_ID
-    CONF_DEVICE_CLASS,
-    CONF_ICON,
-    CONF_NAME,
-    CONF_UNIQUE_ID,
-    CONF_UNIT_OF_MEASUREMENT,
-)
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.exceptions import TemplateError
 from homeassistant.helpers import template
@@ -89,11 +83,6 @@ class TriggerEntity(CoordinatorEntity[TriggerUpdateCoordinator]):
         """Return device class of the entity."""
         return self._config.get(CONF_DEVICE_CLASS)
 
-    @property
-    def unit_of_measurement(self) -> str | None:
-        """Return unit of measurement."""
-        return self._config.get(CONF_UNIT_OF_MEASUREMENT)
-
     @property
     def icon(self) -> str | None:
         """Return icon."""