diff --git a/torch_geometric/nn/models/explainer.py b/torch_geometric/nn/models/explainer.py index a83380d5ed39..12b5c5b767f3 100644 --- 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 index e6393530ddd7..4f4cd332e401 100644 --- 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 index 2ba18b701efc..3ebf33f710d0 100644 --- 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)