style changes

rlratzel · Jan 22, 2020 · 0b360cc · 0b360cc
1 parent eef3c63
commit 0b360cc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/python/cugraph/structure/graph.py b/python/cugraph/structure/graph.py
@@ -222,10 +222,14 @@ def view_edge_list(self):
             if isinstance(self.edgelist.renumber_map, cudf.DataFrame):
                 df = cudf.DataFrame()
                 ncols = len(self.edgelist.edgelist_df) - 2
-                unrenumered_df_ = edgelist_df.merge(self.edgelist.renumber_map, left_on='src', right_on='id', how='left').drop(['id', 'src'])
-                unrenumered_df = unrenumered_df_.merge(self.edgelist.renumber_map, left_on='dst', right_on='id', how='left').drop(['id', 'dst'])
-                cols = unrenumered_df.columns
-                df = unrenumered_df[[cols[ncols:], cols[0:ncols]]]
+                unrnb_df_ = edgelist_df.merge(self.edgelist.renumber_map,
+                                              left_on='src', right_on='id',
+                                              how='left').drop(['id', 'src'])
+                unrnb_df = unrnb_df_.merge(self.edgelist.renumber_map,
+                                           left_on='dst', right_on='id',
+                                           how='left').drop(['id', 'dst'])
+                cols = unrnb_df.columns
+                df = unrnb_df[[cols[ncols:], cols[0:ncols]]]
             else:
                 df = cudf.DataFrame()
                 for col in edgelist_df.columns:

diff --git a/python/cugraph/utilities/unrenumber.py b/python/cugraph/utilities/unrenumber.py
@@ -1,8 +1,11 @@
 import cudf
 
+
 def unrenumber(renumber_map, df, col):
     if isinstance(renumber_map, cudf.DataFrame):
-        unrenumered_df = df.merge(renumber_map, left_on=col, right_on='id', how='left').drop(['id', col])
+        unrenumered_df = df.merge(renumber_map, left_on=col,
+                                  right_on='id',
+                                  how='left').drop(['id', col])
         cols = unrenumered_df.columns
         df = unrenumered_df[[cols[1:], cols[0]]]
     else: