|
1 | 1 | module ADUtils |
2 | 2 |
|
3 | 3 | using ForwardDiff: ForwardDiff |
| 4 | +using Random: Random |
4 | 5 | using ReverseDiff: ReverseDiff |
5 | 6 | using Test: Test |
6 | 7 | using Tracker: Tracker |
@@ -174,7 +175,9 @@ function DynamicPPL.tilde_assume(context::ADTypeCheckContext, right, vn, vi) |
174 | 175 | return value, logp, vi |
175 | 176 | end |
176 | 177 |
|
177 | | -function DynamicPPL.tilde_assume(rng, context::ADTypeCheckContext, sampler, right, vn, vi) |
| 178 | +function DynamicPPL.tilde_assume( |
| 179 | + rng::Random.AbstractRNG, context::ADTypeCheckContext, sampler, right, vn, vi |
| 180 | +) |
178 | 181 | value, logp, vi = DynamicPPL.tilde_assume( |
179 | 182 | rng, DynamicPPL.childcontext(context), sampler, right, vn, vi |
180 | 183 | ) |
@@ -205,7 +208,7 @@ function DynamicPPL.dot_tilde_assume(context::ADTypeCheckContext, right, left, v |
205 | 208 | end |
206 | 209 |
|
207 | 210 | function DynamicPPL.dot_tilde_assume( |
208 | | - rng, context::ADTypeCheckContext, sampler, right, left, vn, vi |
| 211 | + rng::Random.AbstractRNG, context::ADTypeCheckContext, sampler, right, left, vn, vi |
209 | 212 | ) |
210 | 213 | value, logp, vi = DynamicPPL.dot_tilde_assume( |
211 | 214 | rng, DynamicPPL.childcontext(context), sampler, right, left, vn, vi |
|
0 commit comments