|
22 | 22 | DeleteMicrogridDispatchRequest, |
23 | 23 | GetMicrogridDispatchRequest, |
24 | 24 | 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, |
30 | 26 | ListMicrogridDispatchesResponse, |
31 | 27 | UpdateMicrogridDispatchRequest, |
32 | 28 | UpdateMicrogridDispatchResponse, |
@@ -96,7 +92,7 @@ def _check_access(self, metadata: grpc.aio.Metadata) -> None: |
96 | 92 |
|
97 | 93 | # pylint: disable=invalid-name |
98 | 94 | async def ListMicrogridDispatches( |
99 | | - self, request: PBDispatchListRequest, metadata: grpc.aio.Metadata |
| 95 | + self, request: ListMicrogridDispatchesRequest, metadata: grpc.aio.Metadata |
100 | 96 | ) -> ListMicrogridDispatchesResponse: |
101 | 97 | """List microgrid dispatches. |
102 | 98 |
|
@@ -126,7 +122,9 @@ async def ListMicrogridDispatches( |
126 | 122 |
|
127 | 123 | # pylint: disable=too-many-branches |
128 | 124 | @staticmethod |
129 | | - def _filter_dispatch(dispatch: Dispatch, request: PBDispatchListRequest) -> bool: |
| 125 | + def _filter_dispatch( |
| 126 | + dispatch: Dispatch, request: ListMicrogridDispatchesRequest |
| 127 | + ) -> bool: |
130 | 128 | """Filter a dispatch based on the request.""" |
131 | 129 | if request.HasField("filter"): |
132 | 130 | _filter = request.filter |
|
0 commit comments