change to store and show last 20 'remaining terms' in PathInfo

jcmgray · Jul 19, 2020 · 1a80c21 · 1a80c21
1 parent e5d609e
commit 1a80c21
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/opt_einsum/contract.py b/opt_einsum/contract.py
@@ -309,7 +309,9 @@ def contract_path(*operands, **kwargs):
 
         einsum_str = ",".join(tmp_inputs) + "->" + idx_result
 
-        if len(input_list) <= 10:
+        # for large expressions saving the remaining terms at each step can
+        # incur a large memory footprint - and also be messy to print
+        if len(input_list) <= 20:
             remaining = tuple(input_list)
         else:
             remaining = None