Update imports from channels library

Signed-off-by: Simon Völcker <simon.voelcker@frequenz.com>

Author: simonvoelcker

benchmarks/timeseries/benchmark_datasourcing.py

@@ -16,8 +16,13 @@
 from time import perf_counter
 from typing import Any
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, ReceiverStoppedError
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    BroadcastChannel,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    ReceiverStoppedError,
+)
 from frequenz.client.microgrid.metrics import Metric
 from frequenz.quantities import Quantity
 
@@ -78,15 +83,10 @@ async def benchmark_data_sourcing(  # pylint: disable=too-many-locals
     mock_grid.add_ev_chargers(num_ev_chargers)
     mock_grid.start_mock_client(enable_mock_client)
 
-    request_channel = Broadcast[ComponentMetricRequest](
-        name="DataSourcingActor Request Channel"
-    )
-
-    request_receiver = request_channel.new_receiver(
-        name="datasourcing-benchmark",
+    request_sender, request_receiver = BroadcastChannel[ComponentMetricRequest](
+        name="DataSourcingActor Request Channel",
         limit=(num_ev_chargers * len(COMPONENT_METRIC_IDS)),
     )
-    request_sender = request_channel.new_sender()
 
     consume_tasks = []
 

src/frequenz/sdk/microgrid/_data_sourcing/_component_metric_request.py

@@ -6,8 +6,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 
-from frequenz.channels._broadcast import BroadcastReceiver
-from frequenz.channels._oneshot import OneshotSender
+from frequenz.channels import BroadcastReceiver, OneshotSender
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.client.microgrid.metrics import Metric
 from frequenz.quantities import Quantity

src/frequenz/sdk/microgrid/_data_sourcing/microgrid_api_source.py

@@ -8,8 +8,7 @@
 from collections.abc import Callable
 from typing import Any
 
-from frequenz.channels import Broadcast, Receiver
-from frequenz.channels._broadcast import BroadcastSender
+from frequenz.channels import Broadcast, BroadcastSender, Receiver
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.client.microgrid.component import ComponentCategory
 from frequenz.client.microgrid.metrics import Metric

src/frequenz/sdk/microgrid/_power_managing/_base_classes.py

@@ -10,8 +10,7 @@
 import enum
 import typing
 
-from frequenz.channels._broadcast import BroadcastReceiver
-from frequenz.channels._oneshot import OneshotSender
+from frequenz.channels import BroadcastReceiver, OneshotSender
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.quantities import Power
 

src/frequenz/sdk/microgrid/_resampling.py

@@ -7,8 +7,13 @@
 import asyncio
 import logging
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    Broadcast,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    Sender,
+)
 from frequenz.quantities import Quantity
 
 from .._internal._asyncio import cancel_and_await

src/frequenz/sdk/timeseries/_grid_frequency.py

@@ -8,8 +8,13 @@
 import asyncio
 import logging
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    Broadcast,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    Sender,
+)
 from frequenz.channels.experimental import Pipe
 from frequenz.client.microgrid.component import Component, EvCharger, Inverter, Meter
 from frequenz.client.microgrid.metrics import Metric

src/frequenz/sdk/timeseries/_voltage_streamer.py

@@ -13,8 +13,13 @@
 import logging
 from typing import TYPE_CHECKING
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    Broadcast,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    Sender,
+)
 from frequenz.client.microgrid.component import Component, EvCharger, Inverter, Meter
 from frequenz.client.microgrid.metrics import Metric
 from frequenz.quantities import Quantity, Voltage

src/frequenz/sdk/timeseries/battery_pool/_battery_pool.py

@@ -12,8 +12,13 @@
 import uuid
 from collections import abc
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    Broadcast,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    Sender,
+)
 from frequenz.channels.experimental import Pipe
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.quantities import Energy, Percentage, Power, Temperature

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

@@ -8,8 +8,13 @@
 import uuid
 from collections import abc
 
-from frequenz.channels import Broadcast, OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import (
+    Broadcast,
+    BroadcastReceiver,
+    OneshotChannel,
+    Receiver,
+    Sender,
+)
 from frequenz.channels.experimental import Pipe
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.quantities import Current, Power

src/frequenz/sdk/timeseries/formulas/_resampled_stream_fetcher.py

@@ -3,8 +3,7 @@
 
 """Fetches telemetry streams for components."""
 
-from frequenz.channels import OneshotChannel, Receiver, Sender
-from frequenz.channels._broadcast import BroadcastReceiver
+from frequenz.channels import BroadcastReceiver, OneshotChannel, Receiver, Sender
 from frequenz.client.common.microgrid.components import ComponentId
 from frequenz.quantities import Quantity