Fix exception sampling logic

apache · Nov 11, 2024 · a76adea · a76adea
1 parent 2488ca1
commit a76adea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdks/python/apache_beam/runners/common.py b/sdks/python/apache_beam/runners/common.py
@@ -1608,7 +1608,7 @@ def _reraise_augmented(self, exn, windowed_value=None):
     _, _, tb = exc_info
 
     new_exn = new_exn.with_traceback(tb)
-    self._maybe_sample_exception(exc_info, windowed_value)
+    self._maybe_sample_exception(new_exn, windowed_value)
     _LOGGER.exception(new_exn)
     raise new_exn