fix style

rapidsai · Feb 1, 2024 · a6d767f · a6d767f
1 parent 63a669b
commit a6d767f
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py
@@ -464,7 +464,9 @@ def view_edge_list(self):
                 # FIXME: Drop multi edges with the CAPI instead.
                 _client = default_client()
                 workers = _client.scheduler_info()["workers"]
-                edgelist_df = _memory_efficient_drop_duplicates(edgelist_df, [srcCol, dstCol], len(workers))
+                edgelist_df = _memory_efficient_drop_duplicates(
+                    edgelist_df, [srcCol, dstCol], len(workers)
+                )
 
             edgelist_df[srcCol], edgelist_df[dstCol] = edgelist_df[
                 [srcCol, dstCol]

diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py
@@ -264,7 +264,7 @@ def __from_edgelist(
                 source,
                 destination,
                 edge_attr,
-                multi=self.properties.multi_edge, # Deprecated parameter
+                multi=self.properties.multi_edge,  # Deprecated parameter
                 symmetrize=not self.properties.directed,
             )
 
@@ -279,7 +279,7 @@ def __from_edgelist(
                 elist,
                 source,
                 destination,
-                multi=self.properties.multi_edge, # Deprecated parameter
+                multi=self.properties.multi_edge,  # Deprecated parameter
                 symmetrize=not self.properties.directed,
             )
 

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
@@ -141,7 +141,8 @@ def test_mg_uniform_neighbor_sample_simple(dask_client, input_combo):
     vertex_col_name = ["src", "dst"]
     workers = dask_client.scheduler_info()["workers"]
     input_df = _memory_efficient_drop_duplicates(
-        input_df, vertex_col_name, len(workers))
+        input_df, vertex_col_name, len(workers)
+    )
 
     result_nbr = uniform_neighbor_sample(
         dg,