fix felaqua attributes

benleb · Sep 7, 2021 · d68d1c3 · d68d1c3
1 parent 69e88a2
commit d68d1c3
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/sensor.py b/sensor.py
@@ -205,15 +205,12 @@ def state(self) -> float | None:
 
     @property
     def extra_state_attributes(self) -> dict[str, Any]:
-        """Return the remaining water."""
+        """Return additional attrs."""
 
         attrs = {}
 
-        if (
-            state := cast(SureFelaqua, self._coordinator.data[self._id])
-            .raw_data()
-            .get("status")
-        ):
+        if state := cast(SureFelaqua, self._coordinator.data[self._id]).raw_data():
+
             for weight in state.get("drink", {}).get("weights", {}):
                 attr_key = f"weight_{weight['index']}"
                 attrs[attr_key] = weight