From: François Fleuret <francois@fleuret.org>
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}')