format

Author: sbrantq

test/probprog/sample.jl

@@ -24,9 +24,13 @@ end
         seed = Reactant.to_rarray(UInt64[1, 4])
         μ = Reactant.ConcreteRNumber(0.0)
         σ = Reactant.ConcreteRNumber(1.0)
-        before = @code_hlo optimize = false ProbProg.call_internal(one_sample, seed, μ, σ, shape)
+        before = @code_hlo optimize = false ProbProg.call_internal(
+            one_sample, seed, μ, σ, shape
+        )
         @test contains(repr(before), "enzyme.sample")
-        after = @code_hlo optimize = :probprog ProbProg.call_internal(two_samples, seed, μ, σ, shape)
+        after = @code_hlo optimize = :probprog ProbProg.call_internal(
+            two_samples, seed, μ, σ, shape
+        )
         @test !contains(repr(after), "enzyme.sample")
     end
 

test/probprog/simulate.jl

@@ -19,12 +19,12 @@ end
         σ = Reactant.ConcreteRNumber(1.0)
 
         before = @code_hlo optimize = false ProbProg.simulate_internal(
-            model, seed, μ, σ, shape; trace = ProbProg.ProbProgTrace()
+            model, seed, μ, σ, shape; trace=ProbProg.ProbProgTrace()
         )
         @test contains(repr(before), "enzyme.simulate")
 
         after = @code_hlo optimize = :probprog ProbProg.simulate_internal(
-            model, seed, μ, σ, shape; trace = ProbProg.ProbProgTrace()
+            model, seed, μ, σ, shape; trace=ProbProg.ProbProgTrace()
         )
         @test !contains(repr(after), "enzyme.simulate")
     end