@@ -8,7 +8,7 @@ using Statistics: mean
8
8
θ_init = randn (rng, 2 )
9
9
10
10
nuts = NUTS (0.8 )
11
- hmc = HMC (100 ; integrator= Leapfrog (0.05 ))
11
+ hmc = HMC (100 ; integrator = Leapfrog (0.05 ))
12
12
hmcda = HMCDA (0.8 , 0.1 )
13
13
14
14
integrator = Leapfrog (1e-3 )
@@ -38,10 +38,10 @@ using Statistics: mean
38
38
model,
39
39
nuts,
40
40
n_adapts + n_samples;
41
- n_adapts= n_adapts,
42
- initial_params= θ_init,
43
- progress= false ,
44
- verbose= false ,
41
+ n_adapts = n_adapts,
42
+ initial_params = θ_init,
43
+ progress = false ,
44
+ verbose = false ,
45
45
)
46
46
47
47
# Error if keyword argument `nadapts` is used
@@ -50,10 +50,10 @@ using Statistics: mean
50
50
model,
51
51
nuts,
52
52
n_adapts + n_samples;
53
- nadapts= n_adapts,
54
- initial_params= θ_init,
55
- progress= false ,
56
- verbose= false ,
53
+ nadapts = n_adapts,
54
+ initial_params = θ_init,
55
+ progress = false ,
56
+ verbose = false ,
57
57
)
58
58
@test_throws ArgumentError AbstractMCMC. sample (
59
59
rng,
@@ -62,10 +62,10 @@ using Statistics: mean
62
62
MCMCThreads (),
63
63
n_adapts + n_samples,
64
64
2 ;
65
- nadapts= n_adapts,
66
- initial_params= θ_init,
67
- progress= false ,
68
- verbose= false ,
65
+ nadapts = n_adapts,
66
+ initial_params = θ_init,
67
+ progress = false ,
68
+ verbose = false ,
69
69
)
70
70
71
71
# Transform back to original space.
@@ -85,10 +85,10 @@ using Statistics: mean
85
85
model,
86
86
hmc,
87
87
n_adapts + n_samples;
88
- n_adapts= n_adapts,
89
- initial_params= θ_init,
90
- progress= false ,
91
- verbose= false ,
88
+ n_adapts = n_adapts,
89
+ initial_params = θ_init,
90
+ progress = false ,
91
+ verbose = false ,
92
92
)
93
93
94
94
# Transform back to original space.
@@ -108,10 +108,10 @@ using Statistics: mean
108
108
model,
109
109
custom,
110
110
n_adapts + n_samples;
111
- n_adapts= 0 ,
112
- initial_params= θ_init,
113
- progress= false ,
114
- verbose= false ,
111
+ n_adapts = 0 ,
112
+ initial_params = θ_init,
113
+ progress = false ,
114
+ verbose = false ,
115
115
)
116
116
117
117
# Transform back to original space.
@@ -134,20 +134,20 @@ using Statistics: mean
134
134
model,
135
135
custom,
136
136
10 ;
137
- n_adapts= 0 ,
138
- initial_params= θ_init,
139
- progress= false ,
140
- verbose= false ,
137
+ n_adapts = 0 ,
138
+ initial_params = θ_init,
139
+ progress = false ,
140
+ verbose = false ,
141
141
)
142
142
samples2 = AbstractMCMC. sample (
143
143
rng2,
144
144
model,
145
145
custom,
146
146
10 ;
147
- n_adapts= 0 ,
148
- initial_params= θ_init,
149
- progress= false ,
150
- verbose= false ,
147
+ n_adapts = 0 ,
148
+ initial_params = θ_init,
149
+ progress = false ,
150
+ verbose = false ,
151
151
)
152
152
@test mapreduce (* , samples1, samples2) do s1, s2
153
153
s1. z. θ == s2. z. θ
0 commit comments