Set event handler module name (#930)

This commit is contained in:
Nikhil Rao 2023-05-01 21:38:12 -07:00 committed by GitHub
parent 34d77db541
commit ba7225e37c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -98,6 +98,7 @@ class State(Base, ABC, extra=pydantic.Extra.allow):
# Convert the event handlers to functions.
for name, event_handler in self.event_handlers.items():
fn = functools.partial(event_handler.fn, self)
fn.__module__ = event_handler.fn.__module__ # type: ignore
fn.__qualname__ = event_handler.fn.__qualname__ # type: ignore
setattr(self, name, fn)