Skip to content

Commit c096cf6

Browse files
committed
Use the RetryStrategy from client-base
Signed-off-by: Leandro Lucarella <[email protected]>
1 parent ab0f65c commit c096cf6

File tree

3 files changed

+8
-3
lines changed

3 files changed

+8
-3
lines changed

src/frequenz/client/microgrid/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,12 @@
77
"""
88

99

10+
from frequenz.client.base.retry_strategy import (
11+
ExponentialBackoff,
12+
LinearBackoff,
13+
RetryStrategy,
14+
)
15+
1016
from ._client import ApiClient
1117
from ._component import Component
1218
from ._component_data import (
@@ -19,7 +25,6 @@
1925
from ._component_states import EVChargerCableState, EVChargerComponentState
2026
from ._connection import Connection
2127
from ._metadata import Location, Metadata
22-
from ._retry import ExponentialBackoff, LinearBackoff, RetryStrategy
2328

2429
__all__ = [
2530
"ApiClient",

src/frequenz/client/microgrid/_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030

3131
# pylint: enable=no-name-in-module
3232
from frequenz.channels import Broadcast, Receiver, Sender
33+
from frequenz.client.base.retry_strategy import LinearBackoff, RetryStrategy
3334
from google.protobuf.empty_pb2 import Empty # pylint: disable=no-name-in-module
3435

3536
from ._component import (
@@ -49,7 +50,6 @@
4950
from ._connection import Connection
5051
from ._constants import RECEIVER_MAX_SIZE
5152
from ._metadata import Location, Metadata
52-
from ._retry import LinearBackoff, RetryStrategy
5353

5454
DEFAULT_GRPC_CALL_TIMEOUT = 60.0
5555
"""The default timeout for gRPC calls made by this client (in seconds)."""

tests/test_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from frequenz.api.microgrid.microgrid_pb2 import ConnectionFilter as PbConnectionFilter
1919
from frequenz.api.microgrid.microgrid_pb2 import ConnectionList as PbConnectionList
2020
from frequenz.api.microgrid.microgrid_pb2 import SetBoundsParam as PbSetBoundsParam
21+
from frequenz.client.base.retry_strategy import LinearBackoff
2122
from google.protobuf.empty_pb2 import Empty
2223

2324
# pylint: enable=no-name-in-module
@@ -36,7 +37,6 @@
3637
MeterData,
3738
)
3839
from frequenz.client.microgrid._connection import Connection
39-
from frequenz.client.microgrid._retry import LinearBackoff
4040

4141
from . import mock_api
4242

0 commit comments

Comments
 (0)