Skip to content

Commit 90d99ca

Browse files
committed
fix test error
1 parent 41471f6 commit 90d99ca

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

test/ext/DynamicPPLMCMCChainsExt.jl

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,12 @@ end
7575
DynamicPPL.LogDensityFunction(m_lin_reg, DynamicPPL.VarInfo(m_lin_reg)),
7676
AdvancedHMC.NUTS(0.65),
7777
MCMCThreads(),
78-
200,
78+
1000,
7979
2;
8080
chain_type=MCMCChains.Chains,
8181
param_names=[],
82+
discard_initial=100,
83+
n_adapt=100,
8284
)
8385
m_lin_reg_test = linear_reg(xs_test, fill(missing, length(ys_test)))
8486
predictions = DynamicPPL.predict(m_lin_reg_test, chain_lin_reg)
@@ -158,9 +160,10 @@ end
158160
[simple_linear1, simple_linear2, simple_linear3, simple_linear4]
159161
m = model(x, y)
160162
chain = sample(
161-
DynamicPPL.LogDensityFunction(m, DynamicPPL.VarInfo(m)),
163+
DynamicPPL.LogDensityFunction(m),
162164
AdvancedHMC.NUTS(0.65),
163-
1000;
165+
400;
166+
initial_params = rand(4),
164167
chain_type=MCMCChains.Chains,
165168
param_names=param_names[model],
166169
discard_initial=100,

0 commit comments

Comments
 (0)