From 5b1da1f4e2ecb8f29225cdb3ae00300f44dde82e Mon Sep 17 00:00:00 2001
From: osk2 <sc60714@gmail.com>
Date: Mon, 19 Feb 2024 22:02:30 +0800
Subject: [PATCH] Fix device class of humidifier entity

---
 custom_components/panasonic_smart_app/const.py      | 1 +
 custom_components/panasonic_smart_app/humidifier.py | 4 +++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/custom_components/panasonic_smart_app/const.py b/custom_components/panasonic_smart_app/const.py
index 619cc6c..2449e9d 100644
--- a/custom_components/panasonic_smart_app/const.py
+++ b/custom_components/panasonic_smart_app/const.py
@@ -27,6 +27,7 @@
 CONF_UPDATE_INTERVAL = "update_interval"
 
 DEVICE_CLASS_SWITCH = "switch"
+DEVICE_CLASS_DEHUMIDIFIER = "dehumidifier"
 
 DEFAULT_UPDATE_INTERVAL = 180
 
diff --git a/custom_components/panasonic_smart_app/humidifier.py b/custom_components/panasonic_smart_app/humidifier.py
index 3abbce3..ab0a85e 100644
--- a/custom_components/panasonic_smart_app/humidifier.py
+++ b/custom_components/panasonic_smart_app/humidifier.py
@@ -8,6 +8,7 @@
 from .const import (
     DOMAIN,
     DEVICE_TYPE_DEHUMIDIFIER,
+    DEVICE_CLASS_DEHUMIDIFIER,
     DATA_CLIENT,
     DATA_COORDINATOR,
     LABEL_DEHUMIDIFIER,
@@ -112,7 +113,8 @@ def is_on(self) -> bool:
 
     @property
     def device_class(self) -> str:
-        return HumidiferDeviceClass.DEHUMIDIFIER
+        #return HumidiferDeviceClass.DEHUMIDIFIER
+        return DEVICE_CLASS_DEHUMIDIFIER
 
     async def async_set_mode(self, mode) -> None:
         """ Set operation mode """