Fix Tuya vacuum display battery level (home-assistant#61643)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
slemiere · Jan 4, 2022 · 69b1a93 · 69b1a93
1 parent eb91941
commit 69b1a93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/homeassistant/components/tuya/vacuum.py b/homeassistant/components/tuya/vacuum.py
@@ -112,9 +112,9 @@ def __init__(self, device: TuyaDevice, device_manager: TuyaDeviceManager) -> Non
             self._supported_features |= SUPPORT_FAN_SPEED
             self._fan_speed_type = EnumTypeData.from_json(function.values)
 
-        if function := device.function.get(DPCode.ELECTRICITY_LEFT):
+        if status_range := device.status_range.get(DPCode.ELECTRICITY_LEFT):
             self._supported_features |= SUPPORT_BATTERY
-            self._battery_level_type = IntegerTypeData.from_json(function.values)
+            self._battery_level_type = IntegerTypeData.from_json(status_range.values)
 
     @property
     def battery_level(self) -> int | None: