From 74c6bb7c5b7bc96e6d350537073a9ffd2b1f10e7 Mon Sep 17 00:00:00 2001
From: jgrusewski <jgrusewski@localhost>
Date: Wed, 8 Apr 2026 12:24:09 +0200
Subject: [PATCH] fix: update backtest param size test for b0=3 (was 9)

---
 crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs b/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
index 78ba44f8d..e82ceb23e 100644
--- a/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
+++ b/crates/ml/src/cuda_pipeline/gpu_backtest_evaluator.rs
@@ -2078,8 +2078,8 @@ mod tests {
         assert_eq!(sizes[4], 128 * 256);
         // w_v2: 51 * 128 = 6528
         assert_eq!(sizes[6], 51 * 128);
-        // w_b0out: 9 * 51 * 128 = 58752 (9 exposure actions)
-        assert_eq!(sizes[10], 9 * 51 * 128);
+        // w_b0out: 3 * 51 * 128 = 19584 (3 direction actions — was 9 exposure)
+        assert_eq!(sizes[10], 3 * 51 * 128);
         // w_b1out: 3 * 51 * 128 = 19584
         assert_eq!(sizes[14], 3 * 51 * 128);
         // w_b2out: 3 * 51 * 128 = 19584