Skip to content

Commit 923188e

Browse files
fix up tests
1 parent 0a92469 commit 923188e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

test/ensemble.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -83,10 +83,12 @@ sol = solve(enprob; saveat = t_ensem);
8383

8484
@test ensemble_weights(sol, data_ensem) [0.2, 0.5, 0.3]
8585

86-
probs = [prob, prob2, prob3]
87-
ps = [[β => Uniform(0.01, 10.0), γ => Uniform(0.01, 10.0)] for i in 1:3]
88-
datas = [data_train,data_train,data_train]
86+
probs = (prob, prob2, prob3)
87+
ps = Tuple([β => Uniform(0.01, 10.0), γ => Uniform(0.01, 10.0)] for i in 1:3)
88+
datas = (data_train,data_train,data_train)
8989
enprobs = bayesian_ensemble(probs, ps, datas)
9090

9191
sol = solve(enprobs; saveat = t_ensem);
92-
ensemble_weights(sol, data_ensem)
92+
ensemble_weights(sol, data_ensem)
93+
94+
bayesian_datafit(probs, ps, datas)

0 commit comments

Comments
 (0)