Implement missing tilde_assume method (#904)

Author: penelopeysm

HISTORY.md

@@ -1,5 +1,9 @@
 # DynamicPPL Changelog
 
+## 0.36.1
+
+Fixed a missing method for `tilde_assume`.
+
 ## 0.36.0
 
 **Breaking changes**

Project.toml

@@ -1,6 +1,6 @@
 name = "DynamicPPL"
 uuid = "366bfd00-2699-11ea-058f-f148b4cae6d8"
-version = "0.36.0"
+version = "0.36.1"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

src/context_implementations.jl

@@ -57,6 +57,7 @@ function tilde_assume(context::AbstractContext, args...)
     return tilde_assume(NodeTrait(tilde_assume, context), context, args...)
 end
 function tilde_assume(::IsLeaf, context::AbstractContext, right, vn, vi)
+    # no rng nor sampler
     return assume(right, vn, vi)
 end
 function tilde_assume(::IsParent, context::AbstractContext, args...)
@@ -69,11 +70,17 @@ end
 function tilde_assume(
     ::IsLeaf, rng::Random.AbstractRNG, context::AbstractContext, sampler, right, vn, vi
 )
+    # rng and sampler
     return assume(rng, sampler, right, vn, vi)
 end
+function tilde_assume(::IsLeaf, context::AbstractContext, sampler, right, vn, vi)
+    # sampler but no rng
+    return assume(Random.default_rng(), sampler, right, vn, vi)
+end
 function tilde_assume(
     ::IsParent, rng::Random.AbstractRNG, context::AbstractContext, args...
 )
+    # rng but no sampler
     return tilde_assume(rng, childcontext(context), args...)
 end