Adapt _get_zigbee_availability() after rebase

Author: bouwew

plugwise/data.py

@@ -196,7 +196,7 @@ def _get_entity_data(self, entity_id: str, entity: GwEntityData) -> None:
                 entity, "heater_central", "no OpenTherm communication"
             )
         # Zigbee node availability
-        self._get_zigbee_availability(data, entity)
+        self._get_zigbee_availability(entity)
 
         # Switching groups data
         self._entity_switching_group(entity)

plugwise/helper.py

@@ -447,15 +447,13 @@ def _get_measurement_data(self, entity_id: str, entity: GwEntityData) -> None:
 
         entity.update(data)
 
-    def _get_zigbee_availability(
-        self, data: GwEntityData, entity: GwEntityData
-    ) -> None:
+    def _get_zigbee_availability(self, entity: GwEntityData) -> None:
         # Check zigbee device availability
         if "module_id" in entity:
             module_id = entity["module_id"]
             locator = f'./module[@id="{module_id}"]/protocols/zig_bee_node'
             if (module := self._domain_objects.find(locator)) is not None:
-                data["available"] = module.find("reachable").text == "true"
+                entity["available"] = module.find("reachable").text == "true"
 
     def _collect_group_sensors(
         self,