diff --git a/ext/DynamicPPLJETExt.jl b/ext/DynamicPPLJETExt.jl index 143b5a767..fcc6acc0f 100644 --- a/ext/DynamicPPLJETExt.jl +++ b/ext/DynamicPPLJETExt.jl @@ -33,7 +33,7 @@ function DynamicPPL.determine_varinfo( model::DynamicPPL.Model, context::DynamicPPL.AbstractContext=DynamicPPL.DefaultContext(); verbose::Bool=false, - only_tilde::Bool=true + only_tilde::Bool=true, ) # First we try with the typed varinfo. varinfo = DynamicPPL.typed_varinfo(model) @@ -88,4 +88,3 @@ function DynamicPPL.determine_varinfo( end end - diff --git a/test/ext/DynamicPPLJETExt.jl b/test/ext/DynamicPPLJETExt.jl index b04d0177c..e1faa88b4 100644 --- a/test/ext/DynamicPPLJETExt.jl +++ b/test/ext/DynamicPPLJETExt.jl @@ -9,7 +9,8 @@ end end model = demo1() - @test DynamicPPL.determine_varinfo(model; verbose=true) isa DynamicPPL.UntypedVarInfo + @test DynamicPPL.determine_varinfo(model; verbose=true) isa + DynamicPPL.UntypedVarInfo @model demo2() = x ~ Normal() @test DynamicPPL.determine_varinfo(demo2()) isa DynamicPPL.TypedVarInfo @@ -24,7 +25,8 @@ z ~ Normal() end end - @test DynamicPPL.determine_varinfo(demo3(); verbose=true) isa DynamicPPL.UntypedVarInfo + @test DynamicPPL.determine_varinfo(demo3(); verbose=true) isa + DynamicPPL.UntypedVarInfo # Evaluation works (and it would even do so in practice), but sampling # fill fail due to storing `Cauchy{Float64}` in `Vector{Normal{Float64}}`. @@ -36,7 +38,8 @@ y ~ Cauchy() # different distibution, but same transformation => should work end end - @test DynamicPPL.determine_varinfo(demo4(); verbose=true) isa DynamicPPL.UntypedVarInfo + @test DynamicPPL.determine_varinfo(demo4(); verbose=true) isa + DynamicPPL.UntypedVarInfo # In this model, the type error occurs in the user code rather than in DynamicPPL. @model function demo5() @@ -48,9 +51,11 @@ return sum(xs) end # Should pass if we're only checking the tilde statements. - @test DynamicPPL.determine_varinfo(demo5(); verbose=true) isa DynamicPPL.TypedVarInfo + @test DynamicPPL.determine_varinfo(demo5(); verbose=true) isa + DynamicPPL.TypedVarInfo # Should fail if we're including errors in the model body. - @test DynamicPPL.determine_varinfo(demo5(); verbose=true, only_tilde=false) isa DynamicPPL.UntypedVarInfo + @test DynamicPPL.determine_varinfo(demo5(); verbose=true, only_tilde=false) isa + DynamicPPL.UntypedVarInfo end @testset "demo models" begin