Update test_optim.py

liesel-devs · Apr 4, 2024 · 6b2b6f5 · 6b2b6f5
1 parent 070964e
commit 6b2b6f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():