File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 12
12
θ = convert (Vector{Float64}, varinfo[:])
13
13
# Calculate reference logp + gradient of logp using ForwardDiff
14
14
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
+ )
16
18
17
19
@testset " $adtype " for adtype in [
18
20
ADTypes. AutoReverseDiff (; compile= false ),
47
49
x = Vector {T} (undef, TT)
48
50
x[1 ] = α
49
51
for t in 2 : TT
50
- x[t] = x[t- 1 ] + η[t- 1 ] * τ
52
+ x[t] = x[t - 1 ] + η[t - 1 ] * τ
51
53
end
52
54
# measurement model
53
55
y ~ MvNormal (x, σ^ 2 * I)
68
70
spl = Sampler (MyEmptyAlg ())
69
71
vi = VarInfo (model)
70
72
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
72
76
end
73
77
end
You can’t perform that action at this time.
0 commit comments