Skip to content

Commit bfd002a

Browse files
committed
Shorter fix version
1 parent 5968ab6 commit bfd002a

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

test/nn/test_to_hetero_transformer.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -563,9 +563,6 @@ def forward(self, x, edge_index, edge_attr):
563563
idx = torch.arange(data[node_type].num_nodes, device='cuda')
564564
edge_index = torch.stack([idx, idx])
565565
data[(node_type, 'self', node_type)].edge_index = edge_index
566-
data[(node_type, 'self', node_type)].edge_attr = torch.zeros(
567-
idx.numel(), 10, device='cuda'
568-
)
569566

570567
model = to_hetero(Model(), data.metadata())
571568
out_dict = model(data.x_dict, data.edge_index_dict, data.edge_attr_dict)

0 commit comments

Comments
 (0)