Skip to content

Commit 07a077e

Browse files
committed
ActorDispatcher: Make None default identity (id)
Signed-off-by: Mathias L. Baumann <[email protected]>
1 parent bd75c0f commit 07a077e

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/frequenz/dispatch/_actor_dispatcher.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ async def main():
128128
managing_actor = ActorDispatcher(
129129
actor_factory=MyActor.new_with_dispatch,
130130
running_status_receiver=status_receiver,
131-
dispatch_identity=lambda d: d.id,
132131
)
133132
134133
await run(managing_actor)
@@ -139,7 +138,7 @@ def __init__(
139138
self,
140139
actor_factory: Callable[[DispatchInfo, Receiver[DispatchInfo]], Actor],
141140
running_status_receiver: Receiver[Dispatch],
142-
dispatch_identity: Callable[[Dispatch], int],
141+
dispatch_identity: Callable[[Dispatch], int] | None = None,
143142
) -> None:
144143
"""Initialize the dispatch handler.
145144
@@ -151,7 +150,10 @@ def __init__(
151150
By default, it uses the dispatch ID.
152151
"""
153152
super().__init__()
154-
self._dispatch_identity = dispatch_identity
153+
self._dispatch_identity: Callable[[Dispatch], int] = (
154+
dispatch_identity if dispatch_identity else lambda d: d.id
155+
)
156+
155157
self._dispatch_rx = running_status_receiver
156158
self._actor_factory = actor_factory
157159
self._actors: dict[int, Actor] = {}

0 commit comments

Comments
 (0)