Skip to content

Commit 23ae110

Browse files
committed
add source code for MeanField and NoConstraints structures
1 parent 0261b62 commit 23ae110

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

src/plugins/variational_constraints/variational_constraints.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,12 @@ See also: [`BetheFactorization`](@ref)
1717
"""
1818
struct MeanField end
1919

20+
source_code(::MeanField) = "MeanField"
21+
2022
struct NoConstraints end
2123

24+
source_code(::NoConstraints) = "NoConstraints"
25+
2226
include("variational_constraints_macro.jl")
2327
include("variational_constraints_engine.jl")
2428

test/plugins/variational_constraints/variational_constraints_tests.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,4 +1160,10 @@ end
11601160
@test occursin("q(x, y)", source)
11611161
@test occursin("q(x)", source)
11621162
@test occursin("q(y)", source)
1163+
1164+
mf_constraints = GraphPPL.MeanField()
1165+
1166+
source = GraphPPL.source_code(mf_constraints)
1167+
1168+
@test occursin("MeanField", source)
11631169
end

0 commit comments

Comments
 (0)