Resolve ADTypeCheckContext method ambiguity (#2299)

TuringLang · Aug 9, 2024 · 07cc40b · 07cc40b
1 parent 803d2f5
commit 07cc40b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/test_utils/ad_utils.jl b/test/test_utils/ad_utils.jl
@@ -1,6 +1,7 @@
 module ADUtils
 
 using ForwardDiff: ForwardDiff
+using Random: Random
 using ReverseDiff: ReverseDiff
 using Test: Test
 using Tracker: Tracker
@@ -174,7 +175,9 @@ function DynamicPPL.tilde_assume(context::ADTypeCheckContext, right, vn, vi)
     return value, logp, vi
 end
 
-function DynamicPPL.tilde_assume(rng, context::ADTypeCheckContext, sampler, right, vn, vi)
+function DynamicPPL.tilde_assume(
+    rng::Random.AbstractRNG, context::ADTypeCheckContext, sampler, right, vn, vi
+)
     value, logp, vi = DynamicPPL.tilde_assume(
         rng, DynamicPPL.childcontext(context), sampler, right, vn, vi
     )
@@ -205,7 +208,7 @@ function DynamicPPL.dot_tilde_assume(context::ADTypeCheckContext, right, left, v
 end
 
 function DynamicPPL.dot_tilde_assume(
-    rng, context::ADTypeCheckContext, sampler, right, left, vn, vi
+    rng::Random.AbstractRNG, context::ADTypeCheckContext, sampler, right, left, vn, vi
 )
     value, logp, vi = DynamicPPL.dot_tilde_assume(
         rng, DynamicPPL.childcontext(context), sampler, right, left, vn, vi