Allow subclasses of BaseException

apache · Nov 11, 2024 · 76d3d70 · 76d3d70
1 parent a76adea
commit 76d3d70
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/runners/common.py b/sdks/python/apache_beam/runners/common.py
@@ -24,6 +24,7 @@
 
 import collections
 import copy
+import cython
 import logging
 import sys
 import threading
@@ -1503,6 +1504,8 @@ def process(self, windowed_value):
       self._reraise_augmented(exn, windowed_value)
       return []
 
+  # Ignore type annotations to allow subclasses of BaseException
+  @cython.annotation_typing(False)
   def _maybe_sample_exception(
       self, exn: BaseException,
       windowed_value: Optional[WindowedValue]) -> None: