diff --git a/src/context_implementations.jl b/src/context_implementations.jl index bf7291959..c19858816 100644 --- a/src/context_implementations.jl +++ b/src/context_implementations.jl @@ -304,7 +304,9 @@ function dot_tilde_assume(context::PrefixContext, right, left, vn, vi) return dot_tilde_assume(context.context, right, left, prefix.(Ref(context), vn), vi) end -function dot_tilde_assume(rng::Random.AbstractRNG, context::PrefixContext, sampler, right, left, vn, vi) +function dot_tilde_assume( + rng::Random.AbstractRNG, context::PrefixContext, sampler, right, left, vn, vi +) return dot_tilde_assume( rng, context.context, sampler, right, left, prefix.(Ref(context), vn), vi ) @@ -346,7 +348,6 @@ end function dot_tilde_assume( rng::Random.AbstractRNG, context::FixedContext, sampler, right, left, vns, vi ) - if !has_fixed_symbol(context, first(vns)) # Defer to `childcontext`. return tilde_assume(rng, childcontext(context), sampler, right, left, vns, vi) diff --git a/src/contexts.jl b/src/contexts.jl index 11ab2b5a8..bd04d8783 100644 --- a/src/contexts.jl +++ b/src/contexts.jl @@ -504,7 +504,7 @@ setchildcontext(parent::FixedContext, child) = FixedContext(parent.values, child has_fixed_symbol(context::FixedContext, vn::VarName) = has_symbol(context.values, vn) has_symbol(d::AbstractDict, vn::VarName) = haskey(d, vn) -@generated function has_symbol(::NamedTuple{names}, ::VarName{sym}) where {names, sym} +@generated function has_symbol(::NamedTuple{names}, ::VarName{sym}) where {names,sym} return sym in names end diff --git a/test/compiler.jl b/test/compiler.jl index d4f38f5a6..8a11021fe 100644 --- a/test/compiler.jl +++ b/test/compiler.jl @@ -733,7 +733,7 @@ module Issue537 end @testset "invalid .~ expressions" begin @model function demo_with_invalid_dot_tilde() m = Matrix{Float64}(undef, 1, 2) - m .~ [Normal(); Normal()] + return m .~ [Normal(); Normal()] end @test_throws ArgumentError demo_with_invalid_dot_tilde()()