Simplify working batteries check

The _get_working_batteries method had unnecessary complexity
and was redundant as it was returning the original set of batteries
if there was any working battery.
Therefore, this commit removes the _get_working_batteries method and
simplifies the check for working batteries.

Signed-off-by: Daniel Zullo <[email protected]>

Author: daniel-zullo-frequenz

src/frequenz/sdk/actor/power_distributing/power_distributing.py

@@ -537,24 +537,6 @@ def _get_components_pairs(
 
         return bat_inv_map, inv_bat_map
 
-    def _get_working_batteries(self, batteries: Set[int]) -> Set[int]:
-        """Get subset with working batteries.
-
-        If none of the given batteries are working, then treat all of them
-        as working.
-
-        Args:
-            batteries: requested batteries
-
-        Returns:
-            Subset with working batteries or input set if none of the given batteries
-                are working.
-        """
-        working_batteries = self._all_battery_status.get_working_batteries(batteries)
-        if len(working_batteries) == 0:
-            return batteries
-        return working_batteries
-
     def _get_components_data(self, batteries: Set[int]) -> List[InvBatPair]:
         """Get data for the given batteries and adjacent inverters.
 
@@ -568,7 +550,9 @@ def _get_components_data(self, batteries: Set[int]) -> List[InvBatPair]:
             Pairs of battery and adjacent inverter data.
         """
         pairs_data: List[InvBatPair] = []
-        working_batteries = self._get_working_batteries(batteries)
+        working_batteries = (
+            self._all_battery_status.get_working_batteries(batteries) or batteries
+        )
 
         for battery_id in working_batteries:
             if battery_id not in self._battery_receivers: