Skip to content

Commit 8d012f6

Browse files
committed
TestService: Minor refactor of imports
Signed-off-by: Mathias L. Baumann <[email protected]>
1 parent 24a8e3a commit 8d012f6

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

src/frequenz/client/dispatch/test/_service.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,7 @@
2222
DeleteMicrogridDispatchRequest,
2323
GetMicrogridDispatchRequest,
2424
GetMicrogridDispatchResponse,
25-
)
26-
from frequenz.api.dispatch.v1.dispatch_pb2 import (
27-
ListMicrogridDispatchesRequest as PBDispatchListRequest,
28-
)
29-
from frequenz.api.dispatch.v1.dispatch_pb2 import (
25+
ListMicrogridDispatchesRequest,
3026
ListMicrogridDispatchesResponse,
3127
UpdateMicrogridDispatchRequest,
3228
UpdateMicrogridDispatchResponse,
@@ -96,7 +92,7 @@ def _check_access(self, metadata: grpc.aio.Metadata) -> None:
9692

9793
# pylint: disable=invalid-name
9894
async def ListMicrogridDispatches(
99-
self, request: PBDispatchListRequest, metadata: grpc.aio.Metadata
95+
self, request: ListMicrogridDispatchesRequest, metadata: grpc.aio.Metadata
10096
) -> ListMicrogridDispatchesResponse:
10197
"""List microgrid dispatches.
10298
@@ -126,7 +122,9 @@ async def ListMicrogridDispatches(
126122

127123
# pylint: disable=too-many-branches
128124
@staticmethod
129-
def _filter_dispatch(dispatch: Dispatch, request: PBDispatchListRequest) -> bool:
125+
def _filter_dispatch(
126+
dispatch: Dispatch, request: ListMicrogridDispatchesRequest
127+
) -> bool:
130128
"""Filter a dispatch based on the request."""
131129
if request.HasField("filter"):
132130
_filter = request.filter

0 commit comments

Comments
 (0)