fix style

jnke2016 · Oct 2, 2024 · 96d2e20 · 96d2e20
1 parent cfecd90
commit 96d2e20
Showing 1 changed file with 4 additions and 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
@@ -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]]