From a43cd7c4560db7f6f2c73ca3aea41d93e3caef40 Mon Sep 17 00:00:00 2001
From: caiosweet <24454580+caiosweet@users.noreply.github.com>
Date: Wed, 3 May 2023 23:37:55 +0200
Subject: [PATCH] Fixed state class measurement

---
 custom_components/nasa/sensor.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/custom_components/nasa/sensor.py b/custom_components/nasa/sensor.py
index 56465e7..a390355 100644
--- 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,9 +47,9 @@ 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,
     ),
@@ -57,7 +57,7 @@ class NasaSensorEntityDescription(BaseEntityDescription, BaseEntityDescriptionMi
         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]],