Skip to content

Commit f7c8733

Browse files
committed
Formatting. Delete contraction_sequence_to_graph code
1 parent 153d205 commit f7c8733

File tree

6 files changed

+9
-138
lines changed

6 files changed

+9
-138
lines changed

src/ITensorNetworks.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,6 @@ include("formnetworks/abstractformnetwork.jl")
3131
include("formnetworks/bilinearformnetwork.jl")
3232
include("formnetworks/linearformnetwork.jl")
3333
include("formnetworks/quadraticformnetwork.jl")
34-
include("contraction_tree_to_graph.jl")
3534
include("gauging.jl")
3635
include("utils.jl")
3736
include("update_observer.jl")

src/caches/abstractbeliefpropagationcache.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ function region_scalar(bpc::AbstractBeliefPropagationCache, pe::PartitionEdge; k
6666
return not_implemented()
6767
end
6868
partitions(bpc::AbstractBeliefPropagationCache) = not_implemented()
69-
partitionpairs(bpc::AbstractBeliefPropagationCache) = not_implemented()
69+
PartitionedGraphs.partitionedges(bpc::AbstractBeliefPropagationCache) = not_implemented()
7070

7171
function default_edge_sequence(
7272
bpc::AbstractBeliefPropagationCache; alg=default_message_update_alg(bpc)
@@ -107,7 +107,7 @@ function vertex_scalars(bpc::AbstractBeliefPropagationCache, pvs=partitions(bpc)
107107
end
108108

109109
function edge_scalars(
110-
bpc::AbstractBeliefPropagationCache, pes=partitionpairs(bpc); kwargs...
110+
bpc::AbstractBeliefPropagationCache, pes=partitionedges(bpc); kwargs...
111111
)
112112
return map(pe -> region_scalar(bpc, pe; kwargs...), pes)
113113
end
@@ -284,12 +284,12 @@ function update(
284284
return update(Algorithm(alg), bpc; kwargs...)
285285
end
286286

287-
function rescale_message(bp_cache::AbstractBeliefPropagationCache, partitionpair)
288-
return rescale_messages(bp_cache, typeof(partitionpair)[partitionpair])
287+
function rescale_message(bp_cache::AbstractBeliefPropagationCache, partitionedge)
288+
return rescale_messages(bp_cache, typeof(partitionedge)[partitionedge])
289289
end
290290

291291
function rescale_messages(bp_cache::AbstractBeliefPropagationCache)
292-
return rescale_messages(bp_cache, partitionpairs(bp_cache))
292+
return rescale_messages(bp_cache, partitionedges(bp_cache))
293293
end
294294

295295
function rescale_partitions(

src/caches/beliefpropagationcache.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,9 @@ function default_message_update_kwargs(
8080
end
8181

8282
partitions(bpc::BeliefPropagationCache) = partitionvertices(partitioned_tensornetwork(bpc))
83-
partitionpairs(bpc::BeliefPropagationCache) = partitionedges(partitioned_tensornetwork(bpc))
83+
function PartitionedGraphs.partitionedges(bpc::BeliefPropagationCache)
84+
partitionedges(partitioned_tensornetwork(bpc))
85+
end
8486

8587
function set_messages(cache::BeliefPropagationCache, messages)
8688
return BeliefPropagationCache(partitioned_tensornetwork(cache), messages)

src/contraction_tree_to_graph.jl

Lines changed: 0 additions & 82 deletions
This file was deleted.

test/test_belief_propagation.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ using Test: @test, @testset
5050
bpc = BeliefPropagationCache(ψψ, group(v -> first(v), vertices(ψψ)))
5151
bpc = update(bpc; maxiter=25, tol=eps(real(elt)))
5252
#Test messages are converged
53-
for pe in partitionedges(partitioned_tensornetwork(bpc))
53+
for pe in partitionedges(bpc)
5454
@test message_diff(updated_message(bpc, pe), message(bpc, pe)) < 10 * eps(real(elt))
5555
@test eltype(only(message(bpc, pe))) == elt
5656
end

test/test_contraction_sequence_to_graph.jl

Lines changed: 0 additions & 48 deletions
This file was deleted.

0 commit comments

Comments
 (0)