From 8aed649f5e185896b0389a34398fbdb2c6adf3a5 Mon Sep 17 00:00:00 2001
From: Erik <erik@montnemery.com>
Date: Wed, 15 Feb 2023 08:17:22 +0100
Subject: [PATCH] Allow energy, volume and weight sensors to be any state class

---
 homeassistant/components/sensor/const.py | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

diff --git a/homeassistant/components/sensor/const.py b/homeassistant/components/sensor/const.py
index 0727a156d2..ac9f884d27 100644
--- a/homeassistant/components/sensor/const.py
+++ b/homeassistant/components/sensor/const.py
@@ -522,10 +522,7 @@ DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]] = {
     SensorDeviceClass.DATE: set(),
     SensorDeviceClass.DISTANCE: set(SensorStateClass),
     SensorDeviceClass.DURATION: set(SensorStateClass),
-    SensorDeviceClass.ENERGY: {
-        SensorStateClass.TOTAL,
-        SensorStateClass.TOTAL_INCREASING,
-    },
+    SensorDeviceClass.ENERGY: set(SensorStateClass),
     SensorDeviceClass.ENUM: set(),
     SensorDeviceClass.FREQUENCY: {SensorStateClass.MEASUREMENT},
     SensorDeviceClass.GAS: {SensorStateClass.TOTAL, SensorStateClass.TOTAL_INCREASING},
@@ -555,14 +552,11 @@ DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]] = {
     SensorDeviceClass.TIMESTAMP: set(),
     SensorDeviceClass.VOLATILE_ORGANIC_COMPOUNDS: {SensorStateClass.MEASUREMENT},
     SensorDeviceClass.VOLTAGE: {SensorStateClass.MEASUREMENT},
-    SensorDeviceClass.VOLUME: {
-        SensorStateClass.TOTAL,
-        SensorStateClass.TOTAL_INCREASING,
-    },
+    SensorDeviceClass.VOLUME: set(SensorStateClass),
     SensorDeviceClass.WATER: {
         SensorStateClass.TOTAL,
         SensorStateClass.TOTAL_INCREASING,
     },
-    SensorDeviceClass.WEIGHT: {SensorStateClass.TOTAL},
+    SensorDeviceClass.WEIGHT: set(SensorStateClass),
     SensorDeviceClass.WIND_SPEED: {SensorStateClass.MEASUREMENT},
 }