Use ComponentCategory to identify PowerManager type

Earlier a `PoolType` class was used to specify the target component
category the power manager was supposed to support.  That is not
really necessary and the same information can be conveyed with a
component_category argument.

Signed-off-by: Sahas Subramanian <[email protected]>

Author: shsms

src/frequenz/sdk/actor/_power_managing/_power_managing_actor.py

@@ -14,7 +14,8 @@
 from frequenz.channels.util import SkipMissedAndDrift, Timer, select, selected_from
 from typing_extensions import override
 
-from ...timeseries._base_types import PoolType, SystemBounds
+from ...microgrid.component import ComponentCategory
+from ...timeseries._base_types import SystemBounds
 from .._actor import Actor
 from .._channel_registry import ChannelRegistry
 from ._base_classes import Algorithm, BaseAlgorithm, Proposal, ReportRequest, _Report
@@ -31,7 +32,7 @@ class PowerManagingActor(Actor):
 
     def __init__(  # pylint: disable=too-many-arguments
         self,
-        pool_type: PoolType,
+        component_category: ComponentCategory,
         proposals_receiver: Receiver[Proposal],
         bounds_subscription_receiver: Receiver[ReportRequest],
         power_distributing_requests_sender: Sender[power_distributing.Request],
@@ -44,8 +45,8 @@ def __init__(  # pylint: disable=too-many-arguments
         """Create a new instance of the power manager.
 
         Args:
-            pool_type: The type of the component pool this power manager instance is
-                going to support.
+            component_category: The category of the component this power manager
+                instance is going to support.
             proposals_receiver: The receiver for proposals.
             bounds_subscription_receiver: The receiver for bounds subscriptions.
             power_distributing_requests_sender: The sender for power distribution
@@ -63,7 +64,7 @@ def __init__(  # pylint: disable=too-many-arguments
                 f"PowerManagingActor: Unknown algorithm: {algorithm}"
             )
 
-        self._pool_type = pool_type
+        self._component_category = component_category
         self._bounds_subscription_receiver = bounds_subscription_receiver
         self._power_distributing_requests_sender = power_distributing_requests_sender
         self._power_distributing_results_receiver = power_distributing_results_receiver
@@ -132,8 +133,11 @@ def _add_bounds_tracker(self, component_ids: frozenset[int]) -> None:
             microgrid,
         )
 
-        if self._pool_type is not PoolType.BATTERY_POOL:
-            err = f"PowerManagingActor: Unsupported pool type: {self._pool_type}"
+        if self._component_category is not ComponentCategory.BATTERY:
+            err = (
+                "PowerManagingActor: Unsupported component category: "
+                f"{self._component_category}"
+            )
             _logger.error(err)
             raise NotImplementedError(err)
         battery_pool = microgrid.battery_pool(component_ids)

src/frequenz/sdk/microgrid/_power_wrapper.py

@@ -83,10 +83,9 @@ def _start_power_managing_actor(self) -> None:
             return
 
         from ..actor._power_managing._power_managing_actor import PowerManagingActor
-        from ..timeseries._base_types import PoolType
 
         self._power_managing_actor = PowerManagingActor(
-            pool_type=PoolType.BATTERY_POOL,
+            component_category=ComponentCategory.BATTERY,
             proposals_receiver=self.proposal_channel.new_receiver(),
             bounds_subscription_receiver=(
                 self.bounds_subscription_channel.new_receiver()

src/frequenz/sdk/timeseries/_base_types.py

@@ -4,7 +4,6 @@
 """Timeseries basic types."""
 
 import dataclasses
-import enum
 import functools
 from collections.abc import Callable, Iterator
 from dataclasses import dataclass
@@ -172,12 +171,3 @@ class SystemBounds:
     This is the range within which power requests are NOT allowed by the pool.
     If present, they will be a subset of the inclusion bounds.
     """
-
-
-class PoolType(enum.Enum):
-    """Enumeration of component pool types."""
-
-    BATTERY_POOL = "BATTERY_POOL"
-    EV_CHARGER_POOL = "EV_CHARGER_POOL"
-    PV_POOL = "PV_POOL"
-    CHP_POOL = "CHP_POOL"