fix: assert outer exception; not sure how to assert for the original …

…exception
OmicsML · Dec 3, 2023 · f3c3240 · f3c3240
1 parent 09b3bde
commit f3c3240
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/data/test_data.py b/tests/data/test_data.py
@@ -101,7 +101,7 @@ def test_get_data(subtests):
         assert y_test.tolist() == [[2]]
 
         # Validation set not set
-        pytest.raises(KeyError, data.get_val_data)
+        pytest.raises(RuntimeError, data.get_val_data)
 
     with subtests.test("Multi feature"):
         data = Data(adata.copy(), train_size=2)