From: François Fleuret Date: Thu, 8 Dec 2022 16:30:52 +0000 (+0100) Subject: Now also catch ValueError. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=fe350ea557cec0d9f2b9e9c6c88dd9633cd2e879;p=pytorch.git Now also catch ValueError. --- diff --git a/tensorstack.py b/tensorstack.py index 8360f9b..d7b0d89 100755 --- a/tensorstack.py +++ b/tensorstack.py @@ -27,7 +27,7 @@ def exception_hook(exc_type, exc_value, tb): print(f' File "{filename}", line {line_no}, in {name}') print(open(filename, 'r').readlines()[line_no-1]) - if exc_type is RuntimeError: + if exc_type in { RuntimeError, ValueError }: for n,v in tb.tb_frame.f_locals.items(): print(f' {n} -> {v}')