From d3e494aa5c44f14fbe870c43a8e1d37f050fb4e4 Mon Sep 17 00:00:00 2001 From: Angus Hollands Date: Wed, 19 Jul 2023 13:52:03 +0100 Subject: [PATCH] fix: don't decorate non-Exception subclasses (#2587) --- src/awkward/_errors.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/awkward/_errors.py b/src/awkward/_errors.py index ceb9a6ae3a..c861bb49e3 100644 --- a/src/awkward/_errors.py +++ b/src/awkward/_errors.py @@ -59,7 +59,11 @@ def __enter__(self): def __exit__(self, exception_type, exception_value, traceback): try: # Handle caught exception - if exception_type is not None and self.primary() is self: + if ( + exception_type is not None + and issubclass(exception_type, Exception) + and self.primary() is self + ): self.handle_exception(exception_type, exception_value) finally: # `_kwargs` may hold cyclic references, that we really want to avoid