fix loader test

aimclub · May 27, 2024 · e575577 · e575577
1 parent 23e4ad8
commit e575577
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unit/tools/test_load_data.py b/tests/unit/tools/test_load_data.py
@@ -23,8 +23,8 @@ def test_load_multivariate_data():
     train_data, test_data = DataLoader('Epilepsy').load_data()
     x_train, y_train = train_data
     x_test, y_test = test_data
-    assert x_train.shape == (137, 3)
-    assert x_test.shape == (138, 3)
+    assert x_train.shape == (137, 3, 206)
+    assert x_test.shape == (138, 3, 206)
     assert y_train.shape == (137,)
     assert y_test.shape == (138,)