Skip to content

Commit

Permalink
fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
jnke2016 committed Jul 10, 2023
1 parent 2223c75 commit d8af027
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion python/cugraph/cugraph/generators/rmat.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ def _ensure_args_rmat(
raise TypeError("'num_edges' must be an int")
if a + b + c > 1:
raise ValueError("a + b + c should be non-negative and no larger than 1.0")
if not isinstance(clip_and_flip , bool):
if not isinstance(clip_and_flip, bool):
raise TypeError("'clip_and_flip' must be a bool")
if not isinstance(scramble_vertex_ids, bool):
raise TypeError("'scramble_vertex_ids' must be a bool")
Expand Down
8 changes: 4 additions & 4 deletions python/cugraph/cugraph/tests/generators/test_rmat.py
Original file line number Diff line number Diff line change
Expand Up @@ -224,7 +224,9 @@ def test_rmat_edge_ids(scale, include_edge_ids, scramble_vertex_ids):
@pytest.mark.parametrize(
"scramble_vertex_ids", _scramble_vertex_ids, ids=_scramble_vertex_ids_test_ids
)
def test_rmat_edge_types(include_edge_types, min_max_edge_type_value, scramble_vertex_ids):
def test_rmat_edge_types(
include_edge_types, min_max_edge_type_value, scramble_vertex_ids
):
"""
Verifies that the edge types returned by rmat() are valid and that valid values
are passed for 'min_edge_type_value' and 'max_edge_type_value'.
Expand Down Expand Up @@ -286,9 +288,7 @@ def test_rmat_edge_types(include_edge_types, min_max_edge_type_value, scramble_v
"include_edge_weights", _include_edge_weights, ids=_include_edge_weights_test_ids
)
@pytest.mark.parametrize("clip_and_flip", _clip_and_flip, ids=_clip_and_flip_test_ids)
def test_rmat_clip_and_flip(
scale, include_edge_weights, clip_and_flip
):
def test_rmat_clip_and_flip(scale, include_edge_weights, clip_and_flip):
"""
Verifies that there are edges only in the lower triangular part of
the adjacency matrix when 'clip_and_flip' is set to 'true'.
Expand Down

0 comments on commit d8af027

Please sign in to comment.