Skip to content

Commit 6494b34

Browse files
committed
Fix flaky test (I promise it works this time)
1 parent c9d3016 commit 6494b34

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/riemannian.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -490,7 +490,7 @@ end
490490
n::Int,
491491
reps::Int=200,
492492
q::Float64=0.999,
493-
inflate::Float64=1.5,
493+
inflate::Float64=1.8,
494494
rng::AbstractRNG=Random.default_rng(),
495495
)
496496
vals_v = Vector{Float64}(undef, reps)
@@ -537,12 +537,12 @@ end
537537
tols = funnel_w1_tols(; n=n_samples, rng=rng)
538538

539539
@testset "SoftAbsRiemannianMetric" begin
540-
metric = SoftAbsRiemannianMetric((D,), G, ∂G∂θ, 20.0)
540+
metric = SoftAbsRiemannianMetric((D,), G, ∂G∂θ, 40.0)
541541
kinetic = GaussianKinetic()
542542
hamiltonian = Hamiltonian(metric, kinetic, ℓπ, ∂ℓπ∂θ)
543543

544544
initial_ϵ = 0.01
545-
integrator = GeneralizedLeapfrog(initial_ϵ, 15)
545+
integrator = GeneralizedLeapfrog(initial_ϵ, 5)
546546
kernel = HMCKernel(Trajectory{MultinomialTS}(integrator, GeneralisedNoUTurn()))
547547

548548
acceptance_rate = 0.7

0 commit comments

Comments
 (0)