From 742c40c949c8808928657047103a49fc3cdab863 Mon Sep 17 00:00:00 2001
From: arjunsavel <asavel@gmail.com>
Date: Mon, 11 Mar 2024 19:28:08 -0400
Subject: [PATCH] uh change tst

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

diff --git a/src/cortecs/tests/test_integration.py b/src/cortecs/tests/test_integration.py
index 47e853a..eb6a48d 100644
--- a/src/cortecs/tests/test_integration.py
+++ b/src/cortecs/tests/test_integration.py
@@ -202,7 +202,7 @@ def test_optimize(self):
             optimizer.best_params
             == {
                 "n_layers": 2,
-                "n_neurons": 2.0,
+                "n_neurons": 2,
                 "activation": "sigmoid",
                 "learn_rate": 0.01,
             }