formatting

Author: torfjelde

ext/DynamicPPLJETExt.jl

@@ -49,9 +49,7 @@ function DynamicPPL.is_suitable_varinfo(
 end
 
 function DynamicPPL._determine_varinfo_jet(
-    model::DynamicPPL.Model,
-    context::DynamicPPL.AbstractContext;
-    only_tilde::Bool=true,
+    model::DynamicPPL.Model, context::DynamicPPL.AbstractContext; only_tilde::Bool=true
 )
     # First we try with the typed varinfo.
     varinfo = if DynamicPPL.hassampler(context)
@@ -64,9 +62,7 @@ function DynamicPPL._determine_varinfo_jet(
     issuccess = true
 
     # Let's make sure that both evaluation and sampling doesn't result in type errors.
-    issuccess, reports = DynamicPPL.is_suitable_varinfo(
-        model, context, varinfo; only_tilde
-    )
+    issuccess, reports = DynamicPPL.is_suitable_varinfo(model, context, varinfo; only_tilde)
 
     if !issuccess
         # Useful information for debugging.

src/model_utils.jl

@@ -252,9 +252,7 @@ See also: [`DynamicPPL.is_suitable_varinfo`](@ref).
 - `only_tilde`: If `true`, only consider error reports occuring in the tilde pipeline. Default: `true`.
 """
 function determine_suitable_varinfo(
-    model::Model,
-    context::AbstractContext=SamplingContext();
-    only_tilde::Bool=true,
+    model::Model, context::AbstractContext=SamplingContext(); only_tilde::Bool=true
 )
     # If JET.jl has been loaded, and thus `determine_varinfo` has been defined, we use that.
     if Base.get_extension(DynamicPPL, :DynamicPPLJETExt) !== nothing

test/ext/DynamicPPLJETExt.jl

@@ -9,8 +9,7 @@
             end
         end
         model = demo1()
-        @test DynamicPPL.determine_suitable_varinfo(model; verbose=true) isa
-            DynamicPPL.UntypedVarInfo
+        @test DynamicPPL.determine_suitable_varinfo(model) isa DynamicPPL.UntypedVarInfo
 
         @model demo2() = x ~ Normal()
         @test DynamicPPL.determine_suitable_varinfo(demo2()) isa DynamicPPL.TypedVarInfo
@@ -25,8 +24,7 @@
                 z ~ Normal()
             end
         end
-        @test DynamicPPL.determine_suitable_varinfo(demo3(); verbose=true) isa
-            DynamicPPL.UntypedVarInfo
+        @test DynamicPPL.determine_suitable_varinfo(demo3()) 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}}`.
@@ -38,8 +36,7 @@
                 y ~ Cauchy() # different distibution, but same transformation
             end
         end
-        @test DynamicPPL.determine_suitable_varinfo(demo4(); verbose=true) isa
-            DynamicPPL.UntypedVarInfo
+        @test DynamicPPL.determine_suitable_varinfo(demo4()) isa DynamicPPL.UntypedVarInfo
 
         # In this model, the type error occurs in the user code rather than in DynamicPPL.
         @model function demo5()
@@ -51,12 +48,10 @@
             return sum(xs)
         end
         # Should pass if we're only checking the tilde statements.
-        @test DynamicPPL.determine_suitable_varinfo(demo5(); verbose=true) isa
-            DynamicPPL.TypedVarInfo
+        @test DynamicPPL.determine_suitable_varinfo(demo5()) isa DynamicPPL.TypedVarInfo
         # Should fail if we're including errors in the model body.
-        @test DynamicPPL.determine_suitable_varinfo(
-            demo5(); verbose=true, only_tilde=false
-        ) isa DynamicPPL.UntypedVarInfo
+        @test DynamicPPL.determine_suitable_varinfo(demo5(); only_tilde=false) isa
+            DynamicPPL.UntypedVarInfo
     end
 
     @testset "demo models" begin