pep8

pyg-team · Jan 24, 2024 · fbfb709 · fbfb709
1 parent 0c2731b
commit fbfb709
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/torch_geometric/explain/explanation.py b/torch_geometric/explain/explanation.py
@@ -247,7 +247,8 @@ def visualize_graph(self, path: Optional[str] = None,
                 If set to :obj:`None`, will use the most appropriate
                 visualization backend based on available system packages.
                 (default: :obj:`None`)
-            node_labels (List[str], optional): Node labels to use with graphviz.
+            node_labels (List[str], optional): Node labels for graphviz.
+                (default: :obj:`None`)
         """
         edge_mask = self.get('edge_mask')
         if edge_mask is None:

diff --git a/torch_geometric/visualization/graph.py b/torch_geometric/visualization/graph.py
@@ -42,6 +42,8 @@ def visualize_graph(
             If set to :obj:`None`, will use the most appropriate
             visualization backend based on available system packages.
             (default: :obj:`None`)
+        node_labels (List[str], optional): Node labels for graphviz.
+            (default: :obj:`None`)
     """
     if edge_weight is not None:  # Normalize edge weights.
         edge_weight = edge_weight - edge_weight.min()