update networks

Author: GiggleLiu

src/networks.jl

@@ -102,21 +102,10 @@ end
 
 collect_ixs(ne::EinCode) = [collect(ix) for ix in getixs(ne)]
 function collect_ixs(ne::NestedEinsum)
-    d = collect_ixs!(ne, Dict{Int,Vector{OMEinsum.labeltype(ne.eins)}}())
+    d = OMEinsum.collect_ixs!(ne, Dict{Int,Vector{OMEinsum.labeltype(ne.eins)}}())
     return [d[i] for i=1:length(d)]
 end
 
-function collect_ixs!(ne::NestedEinsum, d::Dict)
-    for i=1:length(ne.args)
-        if ne.args[i] isa Integer
-            d[ne.args[i]] = collect(OMEinsum.getixs(ne.eins)[i])
-        else
-            collect_ixs!(ne.args[i], d)
-        end
-    end
-    return d
-end
-
 """
     optimize_code(code; optmethod=:kahypar, sc_target=17, max_group_size=40, nrepeat=10, imbalances=0.0:0.001:0.8, βs=0.01:0.05:10.0, ntrials=50, niters=1000, sc_weight=2.0, rw_weight=1.0)