update doc

pyg-team · Jun 29, 2019 · e908a8a · e908a8a
1 parent 378d474
commit e908a8a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/torch_geometric/utils/convert.py b/torch_geometric/utils/convert.py
@@ -45,7 +45,8 @@ def from_scipy_sparse_matrix(A):
 
 
 def to_networkx(data, node_attrs=None, edge_attrs=None):
-    r"""Converts a data object graph to a networkx graph.
+    r"""Converts a :class:`torch_geometric.data.Data` instance to a
+    :obj:`networkx.DiGraph`.
 
     Args:
         data (torch_geometric.data.Data): The data object.
@@ -73,10 +74,11 @@ def to_networkx(data, node_attrs=None, edge_attrs=None):
 
 
 def from_networkx(G):
-    r""""Converts a networkx graph to a data object graph.
+    r"""Converts a :obj:`networkx.Graph` or :obj:`networkx.DiGraph` to a
+    :class:`torch_geometric.data.Data` instance.
 
     Args:
-        G (networkx.Graph): A networkx graph.
+        G (networkx.Graph or networkx.DiGraph): A networkx graph.
     """
 
     G = G.to_directed() if not nx.is_directed(G) else G