Fix types

CERT-Polska · Mar 25, 2024 · c1dc026 · c1dc026
1 parent 79b6edb
commit c1dc026
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/karton/core/karton.py b/karton/core/karton.py
@@ -6,7 +6,7 @@
 import sys
 import time
 import traceback
-from typing import Any, Callable, Dict, List, Optional, Tuple, cast
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union, cast
 
 from .__version__ import __version__
 from .backend import KartonBackend, KartonBind, KartonMetrics
@@ -290,7 +290,9 @@ def _run_pre_hooks(self) -> None:
                 else:
                     self.log.exception("Pre-hook failed")
 
-    def _run_post_hooks(self, exception: Optional[Exception]) -> None:
+    def _run_post_hooks(
+        self, exception: Optional[Union[Exception, BaseException]]
+    ) -> None:
         """
         Run registered postprocessing hooks