Fix smoke sensor classes

Pirate-Weather · Apr 29, 2024 · 4aadc14 · 4aadc14
1 parent 74940c0
commit 4aadc14
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/custom_components/pirateweather/sensor.py b/custom_components/pirateweather/sensor.py
@@ -376,23 +376,21 @@ class PirateWeatherSensorEntityDescription(SensorEntityDescription):
     "fire_index": PirateWeatherSensorEntityDescription(
         key="fire_index",
         name="Fire Index",
-        state_class=None,
         suggested_display_precision=2,
         icon="mdi:fire",
         forecast_mode=["currently", "hourly"],
     ),
     "fire_index_max": PirateWeatherSensorEntityDescription(
         key="fire_index_max",
         name="Fire Index Max",
-        state_class=None,
         suggested_display_precision=2,
         icon="mdi:fire",
         forecast_mode=["daily"],
     ),
     "smoke": PirateWeatherSensorEntityDescription(
         key="smoke",
         name="Smoke",
-        state_class=SensorDeviceClass.PM25,
+        device_class=SensorDeviceClass.PM25,
         si_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
         us_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
         ca_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
@@ -405,7 +403,7 @@ class PirateWeatherSensorEntityDescription(SensorEntityDescription):
     "smoke_max": PirateWeatherSensorEntityDescription(
         key="smoke_max",
         name="Smoke Max",
-        state_class=SensorDeviceClass.PM25,
+        device_class=SensorDeviceClass.PM25,
         si_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
         us_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
         ca_unit=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,