Skip recording exceptions as return type

data-apis · Aug 31, 2020 · c09670e · c09670e
1 parent cf8c27c
commit c09670e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/record_api/core.py b/record_api/core.py
@@ -418,14 +418,18 @@ def __call__(self) -> None:
 
     def log_called_method(self):
         if self.previous_stack.log_call_args:
+            tos = self.TOS
+            if type(tos) is type and issubclass(tos, Exception):
+                # Don't record exception
+                return
             filename, line, fn, args, *kwargs = self.previous_stack.log_call_args
             kwargs = kwargs[0] if kwargs else {}
             log_call(
                 f"{filename}:{line}",
                 fn,
                 tuple(args),
                 *((kwargs,) if kwargs else ()),
-                return_type=type(self.TOS),
+                return_type=type(tos),
             )
 
     # special case subscr b/c we only check first arg, not both