fix

dudanov · Oct 26, 2024 · 9b63bad · 9b63bad
1 parent 6d2ca22
commit 9b63bad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/custom_components/ftms/sensor.py b/custom_components/ftms/sensor.py
@@ -309,6 +309,10 @@ class FtmsSensorEntity(FtmsEntity, SensorEntity):
     def __init__(self, entry, description) -> None:
         if description.device_class != SensorDeviceClass.ENUM:
             self._attr_native_value = 0
+        elif description.key == TRAINING_STATUS:
+            self._attr_native_value = (
+                entry.runtime_data.ftms.training_status.name.lower()
+            )
 
         super().__init__(entry, description)