From 96d2e2099763c6989122c10217780039164eaa1d Mon Sep 17 00:00:00 2001 From: jnke2016 Date: Wed, 2 Oct 2024 12:47:28 -0700 Subject: [PATCH] fix style --- .../sampling/test_uniform_neighbor_sample_mg.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/python/cugraph/cugraph/tests/sampling/test_uniform_neighbor_sample_mg.py b/python/cugraph/cugraph/tests/sampling/test_uniform_neighbor_sample_mg.py index 81a830a007c..4a85b49a66e 100644 --- a/python/cugraph/cugraph/tests/sampling/test_uniform_neighbor_sample_mg.py +++ b/python/cugraph/cugraph/tests/sampling/test_uniform_neighbor_sample_mg.py @@ -146,10 +146,8 @@ def test_mg_uniform_neighbor_sample_simple(dask_client, input_combo): ) input_df = symmetrize_ddf( - input_df, - src_name="src", - dst_name = "dst", - symmetrize=not dg.is_directed()) + input_df, src_name="src", dst_name="dst", symmetrize=not dg.is_directed() + ) result_nbr = uniform_neighbor_sample( dg, @@ -256,10 +254,8 @@ def test_mg_uniform_neighbor_sample_tree(dask_client, directed): input_df = G.input_df input_df = symmetrize_ddf( - input_df, - src_name="src", - dst_name = "dst", - symmetrize=not G.is_directed()) + input_df, src_name="src", dst_name="dst", symmetrize=not G.is_directed() + ) join = result_nbr.merge( input_df, left_on=[*result_nbr.columns[:2]], right_on=[*input_df.columns[:2]]