From 6b2b6f5b5010ff8ad9a580d20119d4e837808d15 Mon Sep 17 00:00:00 2001
From: Johannes Brachem <37882800+jobrachem@users.noreply.github.com>
Date: Thu, 4 Apr 2024 16:44:30 -0500
Subject: [PATCH] Update test_optim.py

---
 tests/test_optim.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_optim.py b/tests/test_optim.py
index 2af639d..d2212af 100644
--- a/tests/test_optim.py
+++ b/tests/test_optim.py
@@ -200,7 +200,7 @@ def test_history_to_df_pruned(models):
 
     df = history_to_df(result.history["position"])
 
-    assert df.shape == (77, 4)
+    assert df.shape == (84, 4)
 
 
 def test_generate_batches():