FIX: ResLogit test

artefactory · Oct 22, 2024 · be57f67 · be57f67
1 parent 04103f1
commit be57f67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration_tests/models/test_reslogit.py b/tests/integration_tests/models/test_reslogit.py
@@ -232,7 +232,7 @@ def test_reslogit_different_layers_width():
     global dataset
 
     list_n_layers = [0, 1, 3]
-    list_res_layers_width = [[], [], [12, 24, n_items]]
+    list_res_layers_width = [[], [], [12, n_items]]
 
     for n_layers, res_layers_width in zip(list_n_layers, list_res_layers_width):
         model = ResLogit(