From 1652be511e532d5c0dca6efdea3a486e6f8d30cd Mon Sep 17 00:00:00 2001 From: jgrusewski Date: Wed, 8 Apr 2026 02:02:29 +0200 Subject: [PATCH] fix: update test expectations for batch_size=16384 --- crates/ml/src/training_profile.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crates/ml/src/training_profile.rs b/crates/ml/src/training_profile.rs index fecce585a..223bec6e1 100644 --- a/crates/ml/src/training_profile.rs +++ b/crates/ml/src/training_profile.rs @@ -1096,7 +1096,7 @@ mod tests { // [training] values must have been applied assert_eq!(hp.epochs, 200); - assert_eq!(hp.batch_size, 8192); + assert_eq!(hp.batch_size, 16384); assert!((hp.learning_rate - 0.0001).abs() < 1e-10); // OFI MBP-10 data dir should be set in production profile @@ -1364,7 +1364,7 @@ mod tests { // [training] section assert_eq!(hp.epochs, 200); - assert_eq!(hp.batch_size, 8192); + assert_eq!(hp.batch_size, 16384); assert!((hp.learning_rate - 0.0001).abs() < 1e-10); assert!((hp.gamma - 0.99).abs() < 0.001); assert!((hp.reward_scale - 1.0).abs() < 0.01);