fix style

rapidsai · Nov 27, 2023 · 07bcd2e · 07bcd2e
1 parent fdeaa57
commit 07bcd2e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/python/cugraph/cugraph/dask/community/egonet.py b/python/cugraph/cugraph/dask/community/egonet.py
@@ -137,9 +137,7 @@ def ego_graph(input_graph, n, radius=1, center=True):
         n = dask_cudf.from_cudf(n, npartitions=min(input_graph._npartitions, len(n)))
     n = n.astype(n_type)
 
-    n = persist_dask_df_equal_parts_per_worker(
-        n, client, return_type="dict"
-    )
+    n = persist_dask_df_equal_parts_per_worker(n, client, return_type="dict")
     do_expensive_check = False
 
     result = [

diff --git a/python/cugraph/cugraph/dask/community/induced_subgraph.py b/python/cugraph/cugraph/dask/community/induced_subgraph.py
@@ -156,9 +156,7 @@ def induced_subgraph(
         vertices_type = input_graph.input_df.dtypes[0]
 
     if isinstance(vertices, (cudf.Series, cudf.DataFrame)):
-        vertices = dask_cudf.from_cudf(
-            vertices, npartitions=input_graph._npartitions
-        )
+        vertices = dask_cudf.from_cudf(vertices, npartitions=input_graph._npartitions)
     vertices = vertices.astype(vertices_type)
 
     vertices = persist_dask_df_equal_parts_per_worker(

diff --git a/python/cugraph/cugraph/tests/structure/test_graph_mg.py b/python/cugraph/cugraph/tests/structure/test_graph_mg.py
@@ -149,7 +149,8 @@ def test_create_mg_graph(dask_client, input_combo):
     if G.renumbered:
         start = G.lookup_internal_vertex_id(start, None)
     data_start = persist_dask_df_equal_parts_per_worker(
-        start, dask_client, return_type="dict")
+        start, dask_client, return_type="dict"
+    )
 
     res = [
         dask_client.submit(