cleanup to graph classes

rapidsai · Jul 21, 2022 · 4b01c26 · 4b01c26
1 parent f01abaf
commit 4b01c26
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleDistributedGraph.py
@@ -63,7 +63,7 @@ def __init__(self, properties):
         self.source_columns = None
         self.destination_columns = None
 
-    def __make_mg_graph(
+    def _make_plc_graph(
                         sID,
                         edata_x,
                         graph_props,
@@ -195,7 +195,7 @@ def __from_edgelist(
         self._client = default_client()
         self._plc_graph = {
             w: self._client.submit(
-                simpleDistributedGraphImpl.__make_mg_graph,
+                simpleDistributedGraphImpl._make_plc_graph,
                 Comms.get_session_id(),
                 edata,
                 graph_props,

diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py
@@ -200,7 +200,7 @@ def __from_edgelist(
             self._replicate_edgelist()
 
 
-        self.make_plc_graph(value_col=value_col, store_transposed=store_transposed)
+        self._make_plc_graph(value_col=value_col, store_transposed=store_transposed)
 
     def to_pandas_edgelist(self, source='src', destination='dst',
                            weight='weights'):
@@ -761,7 +761,7 @@ def _degree(self, vertex_subset, direction=Direction.ALL):
 
         return df
 
-    def make_plc_graph(self, value_col=None, store_transposed=False):
+    def _make_plc_graph(self, value_col=None, store_transposed=False):
         if value_col is None:
             value_col = cudf.Series(
                 cupy.ones(len(self.edgelist.edgelist_df), dtype='float32')
@@ -807,7 +807,7 @@ def to_directed(self, DiG, store_transposed=False):
         else:
             value_col = None
 
-        DiG.make_plc_graph(value_col, store_transposed)
+        DiG._make_plc_graph(value_col, store_transposed)
 
     def to_undirected(self, G, store_transposed=False):
         """
@@ -836,7 +836,7 @@ def to_undirected(self, G, store_transposed=False):
         else:
             value_col = None
 
-        G.make_plc_graph(value_col, store_transposed)
+        G._make_plc_graph(value_col, store_transposed)
 
     def has_node(self, n):
         """