Skip to content

Commit fc6025a

Browse files
Apply suggestions from code review
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 3f72a46 commit fc6025a

File tree

4 files changed

+12
-5
lines changed

4 files changed

+12
-5
lines changed

ext/DynamicPPLTestExt/contexts.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,9 @@ function DynamicPPL.setchildcontext(
4040
)
4141
return DynamicPPL.TestUtils.TestLogModifyingChildContext(context.mod, child)
4242
end
43-
function DynamicPPL.tilde_assume(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, vn, vi)
43+
function DynamicPPL.tilde_assume(
44+
context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, vn, vi
45+
)
4446
value, logp, vi = DynamicPPL.tilde_assume(context.context, right, vn, vi)
4547
return value, logp * context.mod, vi
4648
end
@@ -50,7 +52,9 @@ function DynamicPPL.dot_tilde_assume(
5052
value, logp, vi = DynamicPPL.dot_tilde_assume(context.context, right, left, vn, vi)
5153
return value, logp * context.mod, vi
5254
end
53-
function DynamicPPL.tilde_observe(context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, left, vi)
55+
function DynamicPPL.tilde_observe(
56+
context::DynamicPPL.TestUtils.TestLogModifyingChildContext, right, left, vi
57+
)
5458
logp, vi = DynamicPPL.tilde_observe(context.context, right, left, vi)
5559
return logp * context.mod, vi
5660
end

ext/DynamicPPLTestExt/sampler.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,8 @@ Test that `sampler` produces the correct marginal posterior means on all models
8181
8282
As of right now, this is just an alias for [`test_sampler_on_demo_models`](@ref).
8383
"""
84-
function DynamicPPL.TestUtils.test_sampler_continuous(sampler::AbstractMCMC.AbstractSampler, args...; kwargs...)
84+
function DynamicPPL.TestUtils.test_sampler_continuous(
85+
sampler::AbstractMCMC.AbstractSampler, args...; kwargs...
86+
)
8587
return test_sampler_on_demo_models(sampler, args...; kwargs...)
8688
end

ext/DynamicPPLTestExt/varinfo.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,9 @@
88
99
Test that `vi[vn]` corresponds to the correct value in `vals` for every `vn` in `vns`.
1010
"""
11-
function DynamicPPL.TestUtils.test_values(vi::AbstractVarInfo, vals::NamedTuple, vns; compare=isequal, kwargs...)
11+
function DynamicPPL.TestUtils.test_values(
12+
vi::AbstractVarInfo, vals::NamedTuple, vns; compare=isequal, kwargs...
13+
)
1214
for vn in vns
1315
@test compare(vi[vn], get(vals, vn); kwargs...)
1416
end

src/test_utils.jl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@ using Bijectors: Bijectors
1010
include("test_utils/model_interface.jl")
1111
include("test_utils/models.jl")
1212

13-
1413
##############################################################
1514
# The remainder of this file contains skeleton implementations for
1615
# DynamicPPLTestExt

0 commit comments

Comments
 (0)