internalize opt_einsum funcs and remove as dep

jcmgray · Aug 11, 2023 · c9c6d3d · c9c6d3d
1 parent e396985
commit c9c6d3d
Show file tree

Hide file tree

Showing 4 changed files with 1,156 additions and 24 deletions.
diff --git a/cotengra/core.py b/cotengra/core.py
@@ -3083,6 +3083,8 @@ def build_divide(
                 partition_opts["imbalance"] = imbalance_s
 
             if dynamic_fix:
+                # for the top level subtree (s==1.0) we partition the outputs
+                # nodes first into their own bi-partition
                 parts_s = 2
                 partition_opts["fix_output_nodes"] = s == 1.0
 
@@ -3114,6 +3116,7 @@ def build_divide(
                 )
                 continue
 
+            # update tree structure with newly contracted subgraphs
             tree.contract_nodes(
                 new_subgs, optimize=super_optimize, check=check
             )