From bf2256ada603b63ddc7cba5df1e809926298ad0a Mon Sep 17 00:00:00 2001
From: Nimrais <29106356+Nimrais@users.noreply.github.com>
Date: Tue, 2 Dec 2025 14:05:34 +0000
Subject: [PATCH] =?UTF-8?q?=F0=9F=A4=96=20Auto-format=20Julia=20code?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 benchmark/continuous_transition_bench.jl      | 41 ++++---------------
 .../continuous_transition.jl                  | 31 ++------------
 2 files changed, 11 insertions(+), 61 deletions(-)

diff --git a/benchmark/continuous_transition_bench.jl b/benchmark/continuous_transition_bench.jl
index f96df7aee..aad4b223a 100644
--- a/benchmark/continuous_transition_bench.jl
+++ b/benchmark/continuous_transition_bench.jl
@@ -54,12 +54,7 @@ function create_benchmark_data(dx, dy)
     m_y = MvNormalMeanCovariance(μy, Σy)
     m_x = MvNormalMeanCovariance(μx, Σx)
 
-    return (
-        meta = meta,
-        q_y = q_y, q_x = q_x, q_a = q_a, q_W = q_W,
-        q_y_x = q_y_x,
-        m_y = m_y, m_x = m_x
-    )
+    return (meta = meta, q_y = q_y, q_x = q_x, q_a = q_a, q_W = q_W, q_y_x = q_y_x, m_y = m_y, m_x = m_x)
 end
 
 # ============================================================================
@@ -67,39 +62,22 @@ end
 # ============================================================================
 
 function bench_a_structured(data)
-    @call_rule ContinuousTransition(:a, Marginalisation) (
-        q_y_x = data.q_y_x,
-        q_a = data.q_a,
-        q_W = data.q_W,
-        meta = data.meta
-    )
+    @call_rule ContinuousTransition(:a, Marginalisation) (q_y_x = data.q_y_x, q_a = data.q_a, q_W = data.q_W, meta = data.meta)
 end
 
 function bench_a_meanfield(data)
-    @call_rule ContinuousTransition(:a, Marginalisation) (
-        q_y = data.q_y,
-        q_x = data.q_x,
-        q_a = data.q_a,
-        q_W = data.q_W,
-        meta = data.meta
-    )
+    @call_rule ContinuousTransition(:a, Marginalisation) (q_y = data.q_y, q_x = data.q_x, q_a = data.q_a, q_W = data.q_W, meta = data.meta)
 end
 
 function bench_marginal_y_x(data)
-    @call_marginalrule ContinuousTransition(:y_x) (
-        m_y = data.m_y,
-        m_x = data.m_x,
-        q_a = data.q_a,
-        q_W = data.q_W,
-        meta = data.meta
-    )
+    @call_marginalrule ContinuousTransition(:y_x) (m_y = data.m_y, m_x = data.m_x, q_a = data.q_a, q_W = data.q_W, meta = data.meta)
 end
 
 # ============================================================================
 #  Benchmark Runner
 # ============================================================================
 
-function run_benchmarks(; quick_mode=false)
+function run_benchmarks(; quick_mode = false)
     println()
     println("=" ^ 80)
     println("  ContinuousTransition Rules Benchmark")
@@ -118,11 +96,7 @@ function run_benchmarks(; quick_mode=false)
     println()
 
     # Results storage
-    results = Dict{String, Vector{Tuple{Int, Int, Float64}}}(
-        "a_structured" => [],
-        "a_meanfield" => [],
-        "marginal_y_x" => []
-    )
+    results = Dict{String, Vector{Tuple{Int, Int, Float64}}}("a_structured" => [], "a_meanfield" => [], "marginal_y_x" => [])
 
     for (dx, dy) in test_dims
         println("-" ^ 60)
@@ -192,5 +166,4 @@ end
 # ============================================================================
 
 quick_mode = length(ARGS) > 0 && ARGS[1] == "quick"
-run_benchmarks(quick_mode=quick_mode)
-
+run_benchmarks(quick_mode = quick_mode)
diff --git a/benchmark/rules/continuous_transition/continuous_transition.jl b/benchmark/rules/continuous_transition/continuous_transition.jl
index bf400af6f..ff056a664 100644
--- a/benchmark/rules/continuous_transition/continuous_transition.jl
+++ b/benchmark/rules/continuous_transition/continuous_transition.jl
@@ -43,12 +43,7 @@ function create_ct_benchmark_data(dx, dy)
     m_y = MvNormalMeanCovariance(μy, Σy)
     m_x = MvNormalMeanCovariance(μx, Σx)
 
-    return (
-        meta = meta,
-        q_y = q_y, q_x = q_x, q_a = q_a, q_W = q_W,
-        q_y_x = q_y_x,
-        m_y = m_y, m_x = m_x
-    )
+    return (meta = meta, q_y = q_y, q_x = q_x, q_a = q_a, q_W = q_W, q_y_x = q_y_x, m_y = m_y, m_x = m_x)
 end
 
 """
@@ -72,23 +67,12 @@ function add_continuous_transition_a_benchmarks(SUITE)
 
         # Structured VMP: q(y,x) joint
         SUITE["a"]["Structured"]["dx=$(dx), dy=$(dy)"] = @benchmarkable begin
-            @call_rule ContinuousTransition(:a, Marginalisation) (
-                q_y_x = $data.q_y_x,
-                q_a = $data.q_a,
-                q_W = $data.q_W,
-                meta = $data.meta
-            )
+            @call_rule ContinuousTransition(:a, Marginalisation) (q_y_x = $data.q_y_x, q_a = $data.q_a, q_W = $data.q_W, meta = $data.meta)
         end
 
         # Mean-field VMP: q(y)q(x)q(a)q(W)
         SUITE["a"]["Mean-field"]["dx=$(dx), dy=$(dy)"] = @benchmarkable begin
-            @call_rule ContinuousTransition(:a, Marginalisation) (
-                q_y = $data.q_y,
-                q_x = $data.q_x,
-                q_a = $data.q_a,
-                q_W = $data.q_W,
-                meta = $data.meta
-            )
+            @call_rule ContinuousTransition(:a, Marginalisation) (q_y = $data.q_y, q_x = $data.q_x, q_a = $data.q_a, q_W = $data.q_W, meta = $data.meta)
         end
     end
 end
@@ -104,14 +88,7 @@ function add_continuous_transition_marginals_benchmarks(SUITE)
 
         # y_x marginal rule
         SUITE["marginals"]["y_x"]["dx=$(dx), dy=$(dy)"] = @benchmarkable begin
-            @call_marginalrule ContinuousTransition(:y_x) (
-                m_y = $data.m_y,
-                m_x = $data.m_x,
-                q_a = $data.q_a,
-                q_W = $data.q_W,
-                meta = $data.meta
-            )
+            @call_marginalrule ContinuousTransition(:y_x) (m_y = $data.m_y, m_x = $data.m_x, q_a = $data.q_a, q_W = $data.q_W, meta = $data.meta)
         end
     end
 end
-