Fix type exception type hints

pyflowlauncher/event.py

@@ -1,6 +1,6 @@
 
 
-from typing import Any, Callable, Iterable
+from typing import Any, Callable, Iterable, Type
 
 
 class EventHandler:
@@ -21,7 +21,7 @@ def add_methods(self, methods: Iterable[Callable[..., Any]]):
         for method in methods:
             self.add_method(method)
 
-    def add_exception_handler(self, exception: Exception, handler: Callable[..., Any]):
+    def add_exception_handler(self, exception: Type[Exception], handler: Callable[..., Any]):
         self._handlers[exception] = handler
 
     def __call__(self, method: str, *args, **kwargs):

pyflowlauncher/plugin.py

@@ -2,7 +2,7 @@
 
 import sys
 from functools import wraps
-from typing import Any, Callable, Iterable, Optional, Union
+from typing import Any, Callable, Iterable, Optional, Type, Union
 from pathlib import Path
 import json
 
@@ -44,7 +44,7 @@ def method(self, method: Method) -> Method:
         """Register a method to be called when the plugin is run."""
         return self.on_method(method)
 
-    def add_exception_handler(self, exception: Exception, handler: Callable[..., Any]) -> None:
+    def add_exception_handler(self, exception: Type[Exception], handler: Callable[..., Any]) -> None:
         """Add exception handler to be called when an exception is raised in a method."""
         self._event_handler.add_exception_handler(exception, handler)