Merge pull request #7 from artemyfmv/fix-app-module-type-annotations

Fix type annotations in `ApplicationModule` class

lato/application_module.py

@@ -17,7 +17,9 @@ def __init__(self, name: str):
         :param name: Name of the module
         """
         self.name: str = name
-        self._handlers: defaultdict[str, OrderedSet[Callable]] = defaultdict(OrderedSet)
+        self._handlers: defaultdict[HandlerAlias, OrderedSet[Callable]] = defaultdict(
+            OrderedSet
+        )
         self._submodules: OrderedSet[ApplicationModule] = OrderedSet()
 
     @property
@@ -34,7 +36,7 @@ def include_submodule(self, a_module: "ApplicationModule"):
         ), f"Can only include {ApplicationModule} instances, got {a_module}"
         self._submodules.add(a_module)
 
-    def handler(self, alias: HandlerAlias):
+    def handler(self, alias: HandlerAlias) -> Callable:
         """
         Decorator for registering a handler. Handler can be aliased by a name or by a message type.