Fix some typos in doc (#5451)

pyg-team · Sep 15, 2022 · 36244de · 36244de
1 parent 66c05ec
commit 36244de
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/torch_geometric/nn/models/explainer.py b/torch_geometric/nn/models/explainer.py
@@ -95,7 +95,7 @@ def to_captum(model: torch.nn.Module, mask_type: str = "edge",
     .. code-block:: python
 
         from captum.attr import IntegratedGradients
-        from torch_geometric.nn import GCN, from_captum
+        from torch_geometric.nn import GCN, to_captum
 
         model = GCN(...)
         ...  # Train the model.

diff --git a/torch_geometric/nn/to_hetero_transformer.py b/torch_geometric/nn/to_hetero_transformer.py
@@ -39,6 +39,7 @@ def to_hetero(module: Module, metadata: Metadata, aggr: str = "sum",
 
         class GNN(torch.nn.Module):
             def __init__(self):
+                super().__init__()
                 self.conv1 = SAGEConv((-1, -1), 32)
                 self.conv2 = SAGEConv((32, 32), 32)
 
@@ -52,7 +53,7 @@ def forward(self, x, edge_index):
         node_types = ['paper', 'author']
         edge_types = [
             ('paper', 'cites', 'paper'),
-            ('paper' 'written_by', 'author'),
+            ('paper', 'written_by', 'author'),
             ('author', 'writes', 'paper'),
         ]
         metadata = (node_types, edge_types)

diff --git a/torch_geometric/nn/to_hetero_with_bases_transformer.py b/torch_geometric/nn/to_hetero_with_bases_transformer.py
@@ -43,6 +43,7 @@ def to_hetero_with_bases(module: Module, metadata: Metadata, num_bases: int,
 
         class GNN(torch.nn.Module):
             def __init__(self):
+                super().__init__()
                 self.conv1 = SAGEConv((16, 16), 32)
                 self.conv2 = SAGEConv((32, 32), 32)
 
@@ -56,7 +57,7 @@ def forward(self, x, edge_index):
         node_types = ['paper', 'author']
         edge_types = [
             ('paper', 'cites', 'paper'),
-            ('paper' 'written_by', 'author'),
+            ('paper', 'written_by', 'author'),
             ('author', 'writes', 'paper'),
         ]
         metadata = (node_types, edge_types)