Fixed state class measurement

caiosweet · May 3, 2023 · a43cd7c · a43cd7c
1 parent ec84336
commit a43cd7c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/custom_components/nasa/sensor.py b/custom_components/nasa/sensor.py
@@ -8,7 +8,7 @@
 from homeassistant.components.sensor import (
     SensorEntity,
     SensorEntityDescription,
-    SensorStateClass,
+    # SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
@@ -47,17 +47,17 @@ class NasaSensorEntityDescription(BaseEntityDescription, BaseEntityDescriptionMi
         key="apod",
         name="APOD",
         icon="mdi:image",
-        native_unit_of_measurement="Apod",
+        # native_unit_of_measurement="Apod",
         entity_category=EntityCategory.DIAGNOSTIC,
-        state_class=SensorStateClass.MEASUREMENT,
+        # state_class=SensorStateClass.MEASUREMENT,
         value_fn=lambda data: data["title"],
         attr_fn=lambda data: data,
     ),
     NasaSensorEntityDescription(
         key="epic",
         name="EPIC",
         icon="mdi:earth",
-        state_class=SensorStateClass.MEASUREMENT,
+        # state_class=SensorStateClass.MEASUREMENT,
         # avabl_fn=lambda data: data["date"],
         value_fn=lambda data: data["last_date"],
         attr_fn=lambda data: data,
@@ -67,7 +67,7 @@ class NasaSensorEntityDescription(BaseEntityDescription, BaseEntityDescriptionMi
         key="insight",
         name="InSight",
         icon="mdi:sun-compass",
-        state_class=SensorStateClass.MEASUREMENT,
+        # state_class=SensorStateClass.MEASUREMENT,
         avabl_fn=lambda data: data["sol_keys"],
         value_fn=lambda data: data["sol_keys"][-1],
         attr_fn=lambda data: data[data["sol_keys"][-1]],