Skip to content

Commit 13fa184

Browse files
Format
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 5be363f commit 13fa184

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

test/ad.jl

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@
1212
θ = convert(Vector{Float64}, varinfo[:])
1313
# Calculate reference logp + gradient of logp using ForwardDiff
1414
default_adtype = ADTypes.AutoForwardDiff(; chunksize=0)
15-
ref_logp, ref_grad = LogDensityProblems.logdensity_and_gradient(f, θ, default_adtype)
15+
ref_logp, ref_grad = LogDensityProblems.logdensity_and_gradient(
16+
f, θ, default_adtype
17+
)
1618

1719
@testset "$adtype" for adtype in [
1820
ADTypes.AutoReverseDiff(; compile=false),
@@ -47,7 +49,7 @@
4749
x = Vector{T}(undef, TT)
4850
x[1] = α
4951
for t in 2:TT
50-
x[t] = x[t-1] + η[t-1] * τ
52+
x[t] = x[t - 1] + η[t - 1] * τ
5153
end
5254
# measurement model
5355
y ~ MvNormal(x, σ^2 * I)
@@ -68,6 +70,8 @@
6870
spl = Sampler(MyEmptyAlg())
6971
vi = VarInfo(model)
7072
ldf = DynamicPPL.LogDensityFunction(vi, model, SamplingContext(spl))
71-
@test LogDensityProblems.logdensity_and_gradient(ldf, vi[:], AutoReverseDiff(; compile=true)) isa Any
73+
@test LogDensityProblems.logdensity_and_gradient(
74+
ldf, vi[:], AutoReverseDiff(; compile=true)
75+
) isa Any
7276
end
7377
end

0 commit comments

Comments
 (0)