diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py index 2e43ba8..e857402 100644 --- a/more_math/Parser/UnifiedMathVisitor.py +++ b/more_math/Parser/UnifiedMathVisitor.py @@ -362,6 +362,7 @@ class UnifiedMathVisitor(MathExprVisitor): self._owned_tensors = weakref.WeakValueDictionary() def visit(self, tree): + throw_exception_if_processing_interrupted() if tree is None: return None