Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
TuringLang · Nov 1, 2024 · e66e6ae · e66e6ae
1 parent 0b7ba4b
commit e66e6ae
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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()()