Default partitioning - default partitioned vertices

ITensor · Mar 21, 2024 · 1d04a04 · 1d04a04
1 parent 70258b4
commit 1d04a04
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/caches/beliefpropagationcache.jl b/src/caches/beliefpropagationcache.jl
@@ -10,7 +10,7 @@ end
 @traitfn function default_bp_maxiter(g::::IsDirected)
   return default_bp_maxiter(undirected_graph(underlying_graph(g)))
 end
-default_partitioning(ψ::AbstractITensorNetwork) = group(v -> v, vertices(ψ))
+default_partitioned_vertices(ψ::AbstractITensorNetwork) = group(v -> v, vertices(ψ))
 default_cache_update_kwargs(cache) = (; maxiter=20, tol=1e-5)
 
 function message_diff(message_a::Vector{ITensor}, message_b::Vector{ITensor})

diff --git a/src/environment.jl b/src/environment.jl
@@ -26,12 +26,12 @@ function environment(
   ψ::AbstractITensorNetwork,
   verts::Vector;
   (cache!)=nothing,
-  partitions=default_partitioning(ψ),
+  partitioned_vertices=default_partitioned_vertices(ψ),
   update_cache=isnothing(cache!),
   cache_update_kwargs=default_cache_update_kwargs(cache!),
 )
   if isnothing(cache!)
-    cache! = Ref(BeliefPropagationCache(ψ, partitions))
+    cache! = Ref(BeliefPropagationCache(ψ, partitioned_vertices))
   end
 
   if update_cache