diff --git a/python/cugraph/cugraph/tests/link_prediction/test_overlap.py b/python/cugraph/cugraph/tests/link_prediction/test_overlap.py index 11ef0047b63..4b00330b6c9 100644 --- a/python/cugraph/cugraph/tests/link_prediction/test_overlap.py +++ b/python/cugraph/cugraph/tests/link_prediction/test_overlap.py @@ -182,7 +182,9 @@ def test_directed_graph_check(graph_file, use_weight): vertex_pair = vertex_pair[:5] with pytest.raises(ValueError): - cugraph.overlap(G1, vertex_pair, use_weight) + cugraph.overlap( + G1, vertex_pair, do_expensive_check=False, use_weight=use_weight + ) @pytest.mark.sg diff --git a/python/cugraph/cugraph/tests/link_prediction/test_sorensen.py b/python/cugraph/cugraph/tests/link_prediction/test_sorensen.py index 8806f135302..6345187a376 100644 --- a/python/cugraph/cugraph/tests/link_prediction/test_sorensen.py +++ b/python/cugraph/cugraph/tests/link_prediction/test_sorensen.py @@ -219,7 +219,9 @@ def test_directed_graph_check(read_csv, use_weight): vertex_pair = vertex_pair[:5] with pytest.raises(ValueError): - cugraph.sorensen(G1, vertex_pair, use_weight) + cugraph.sorensen( + G1, vertex_pair, do_expensive_check=False, use_weight=use_weight + ) @pytest.mark.sg