uh change tst

arjunsavel · Mar 11, 2024 · 742c40c · 742c40c
1 parent 2852799
commit 742c40c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
             }