Skip to content

Commit adeaac4

Browse files
committed
Replace Merge with merge for constructing and Merger for typing
Signed-off-by: Sahas Subramanian <[email protected]>
1 parent 2a6c2d1 commit adeaac4

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,7 @@
1010
from collections import abc
1111
from datetime import timedelta
1212

13-
from frequenz.channels import Broadcast, Receiver, Sender
14-
from frequenz.channels.util import Merge
13+
from frequenz.channels import Broadcast, Merger, Receiver, Sender, merge
1514

1615
from ..._internal._asyncio import cancel_and_await
1716
from ._component_status import (
@@ -88,7 +87,7 @@ async def stop(self) -> None:
8887

8988
def _make_merged_status_receiver(
9089
self,
91-
) -> Merge[ComponentStatus]:
90+
) -> Merger[ComponentStatus]:
9291
status_receivers: list[Receiver[ComponentStatus]] = []
9392

9493
for component_id in self._component_ids:
@@ -104,7 +103,7 @@ def _make_merged_status_receiver(
104103
)
105104
self._component_status_trackers.append(tracker)
106105
status_receivers.append(channel.new_receiver())
107-
return Merge(*status_receivers)
106+
return merge(*status_receivers)
108107

109108
async def _run(self) -> None:
110109
"""Start tracking component status."""

src/frequenz/sdk/timeseries/ev_charger_pool/_state_tracker.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,7 @@
88
import asyncio
99
from enum import Enum
1010

11-
from frequenz.channels import Receiver
12-
from frequenz.channels.util import Merge
11+
from frequenz.channels import Merger, Receiver, merge
1312
from frequenz.client.microgrid import (
1413
EVChargerCableState,
1514
EVChargerComponentState,
@@ -87,7 +86,7 @@ def __init__(self, component_ids: set[int]) -> None:
8786
"""
8887
self._component_ids = component_ids
8988
self._task: asyncio.Task[None] = asyncio.create_task(self._run())
90-
self._merged_stream: Merge[EVChargerData] | None = None
89+
self._merged_stream: Merger[EVChargerData] | None = None
9190

9291
# Initialize all components to the `MISSING` state. This will change as data
9392
# starts arriving from the individual components.
@@ -124,7 +123,7 @@ async def _run(self) -> None:
124123
streams: list[Receiver[EVChargerData]] = await asyncio.gather(
125124
*[api_client.ev_charger_data(cid) for cid in self._component_ids]
126125
)
127-
self._merged_stream = Merge(*streams)
126+
self._merged_stream = merge(*streams)
128127
async for data in self._merged_stream:
129128
self._update(data)
130129

0 commit comments

Comments
 (0)