Imported Algorithmn from ITensors

ITensor · Nov 9, 2023 · b21f629 · b21f629
1 parent fe06b39
commit b21f629
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 6 deletions.
diff --git a/examples/belief_propagation/bpsequences.jl b/examples/belief_propagation/bpsequences.jl
@@ -53,7 +53,7 @@ function main()
       contract_kwargs=(; alg="exact"),
       target_precision=1e-10,
       niters=100,
-      edges=edge_sequence(mts_init; alg=ITensors.Algorithm("Parallel")),
+      edges=edge_sequence(mts_init; alg="parallel"),
       verbose=true,
     )
     print("Sequential updates (sequence is default edge list of the message tensors): ")

diff --git a/examples/gauging/gauging_itns.jl b/examples/gauging/gauging_itns.jl
@@ -75,10 +75,7 @@ function benchmark_state_gauging(
         )
       else
         times_iters[i] = @elapsed mts, _ = belief_propagation_iteration(
-          ψψ,
-          mts;
-          contract_kwargs=(; alg="exact"),
-          edges=edge_sequence(mts; alg=ITensors.Algorithm("Parallel")),
+          ψψ, mts; contract_kwargs=(; alg="exact"), edges=edge_sequence(mts; alg="parallel")
         )
       end
 

diff --git a/src/beliefpropagation/beliefpropagation_schedule.jl b/src/beliefpropagation/beliefpropagation_schedule.jl
@@ -1,4 +1,4 @@
-default_edge_sequence_alg() = Algorithm("ForestCover")
+default_edge_sequence_alg() = "forest_cover"
 
 default_bp_niters(g::NamedGraph) = is_tree(g) ? 1 : nothing
 function default_bp_niters(g::AbstractGraph)

diff --git a/src/imports.jl b/src/imports.jl
@@ -96,6 +96,8 @@ import ITensors:
   scalartype,
   #adding
   add
+#Algorithm
+Algorithm
 
 using ITensors.ContractionSequenceOptimization: deepmap