formatting

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

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git 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
@@ -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