Skip to content

Commit c8f4422

Browse files
committed
simplify test info
1 parent fe67fa7 commit c8f4422

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/hmc.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,15 +10,15 @@ n_adapts = 1_000
1010

1111
@testset "HMC and NUTS" begin
1212
temp = randn(D,100)
13-
@testset "$metric" for metric in [UnitEuclideanMetric(θ_init),
13+
@testset "$typeof(metric)" for metric in [UnitEuclideanMetric(θ_init),
1414
DiagEuclideanMetric(θ_init, vec(var(temp; dims=2))),
1515
DenseEuclideanMetric(θ_init, cov(temp'))]
1616
h = Hamiltonian(metric, logπ, ∂logπ∂θ)
17-
@testset "$prop" for prop in [TakeLastProposal(Leapfrog(ϵ), n_steps),
17+
@testset "$typeof(prop)" for prop in [TakeLastProposal(Leapfrog(ϵ), n_steps),
1818
SliceNUTS(Leapfrog(find_good_eps(h, θ_init)))]
1919
samples = AdvancedHMC.sample(h, prop, θ_init, n_samples; verbose=false)
2020
@test mean(samples) zeros(D) atol=RNDATOL
21-
@testset "$adapter" for adapter in [PreConditioner(metric), DualAveraging(0.8, prop.integrator.ϵ),
21+
@testset "$typeof(adapter)" for adapter in [PreConditioner(metric), DualAveraging(0.8, prop.integrator.ϵ),
2222
NaiveCompAdapter(PreConditioner(metric), DualAveraging(0.8, prop.integrator.ϵ)),
2323
ThreePhaseAdapter(n_adapts, PreConditioner(metric), DualAveraging(0.8, prop.integrator.ϵ))]
2424
samples = AdvancedHMC.sample(h, prop, θ_init, n_samples, adapter, n_adapts; verbose=false)

0 commit comments

Comments
 (0)