From 88302d2fbe5ca4adc72f24a78e8d2abb5326418c Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 5 Dec 2022 09:42:48 +0100 Subject: [PATCH] Update. --- tensorstack.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensorstack.py b/tensorstack.py index f511547..8360f9b 100755 --- a/tensorstack.py +++ b/tensorstack.py @@ -20,17 +20,18 @@ def exception_hook(exc_type, exc_value, tb): Tensor.__repr__=lambda x: f'{x.size()}:{x.dtype}:{x.device}' while tb: - print('--------------------------------------------------') + print('--------------------------------------------------\n') filename = tb.tb_frame.f_code.co_filename name = tb.tb_frame.f_code.co_name line_no = tb.tb_lineno print(f' File "{filename}", line {line_no}, in {name}') - print(open(filename, 'r').readlines()[line_no-1], end='') + print(open(filename, 'r').readlines()[line_no-1]) if exc_type is RuntimeError: for n,v in tb.tb_frame.f_locals.items(): print(f' {n} -> {v}') + print() tb = tb.tb_next Tensor.__repr__=repr_orig -- 2.20.1