Skip to content

Commit 2be31b4

Browse files
committed
tidy tests advi
1 parent 1b2351f commit 2be31b4

File tree

1 file changed

+4
-16
lines changed

1 file changed

+4
-16
lines changed

test/variational/advi.jl

Lines changed: 4 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -51,14 +51,8 @@ using Turing.Variational
5151
end
5252

5353
@testset "custom algorithm $name" for (name, algorithm) in [
54-
(
55-
"KLMinRepGradProxDescent",
56-
KLMinRepGradProxDescent(AutoReverseDiff(); n_samples=10),
57-
),
58-
(
59-
"KLMinRepGradDescent",
60-
KLMinRepGradDescent(AutoReverseDiff(); operator, n_samples=10),
61-
),
54+
("KLMinRepGradProxDescent", KLMinRepGradProxDescent(adtype; n_samples=10)),
55+
("KLMinRepGradDescent", KLMinRepGradDescent(adtype; operator, n_samples=10)),
6256
]
6357
T = 1000
6458
q, _, _ = vi(
@@ -74,14 +68,8 @@ using Turing.Variational
7468
end
7569

7670
@testset "inference $name" for (name, algorithm) in [
77-
(
78-
"KLMinRepGradProxDescent",
79-
KLMinRepGradProxDescent(AutoReverseDiff(); n_samples=10),
80-
),
81-
(
82-
"KLMinRepGradDescent",
83-
KLMinRepGradDescent(AutoReverseDiff(); operator, n_samples=10),
84-
),
71+
("KLMinRepGradProxDescent", KLMinRepGradProxDescent(adtype; n_samples=10)),
72+
("KLMinRepGradDescent", KLMinRepGradDescent(adtype; operator, n_samples=10)),
8573
]
8674
rng = StableRNG(0x517e1d9bf89bf94f)
8775

0 commit comments

Comments
 (0)