diff --git a/prettytensor/pretty_tensor_class.py b/prettytensor/pretty_tensor_class.py index a96c693..d701cec 100644 --- a/prettytensor/pretty_tensor_class.py +++ b/prettytensor/pretty_tensor_class.py @@ -1332,7 +1332,8 @@ def _strip_unnecessary_contents_from_stack(result, processed): processed.add(result) trace = [] found = False - for f, line_no, method, _ in result._traceback: + for trace_content in result._traceback: + f, line_no, method = trace_content[0:3] if (method in ('_replace_deferred', '_construct') and f.endswith('pretty_tensor_class.py')): found = True