fixed Error type assertion in modification test

florencejt · Nov 29, 2023 · 0a99bcf · 0a99bcf
1 parent f20a939
commit 0a99bcf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_models/test_subspace_and_graph_methods.py b/tests/test_models/test_subspace_and_graph_methods.py
@@ -499,7 +499,7 @@ def test_AttentionWeightedGraphMaker():
     assert hasattr(attention_weighted_graph_maker, "early_stop_callback")
 
     # Check if parameters are valid
-    with pytest.raises(ValueError):
+    with pytest.raises(TypeError):
         attention_weighted_graph_maker.edge_probability_threshold = -0.1
         attention_weighted_graph_maker.check_params()