GSN_sparse.py

1 F401: 'torch_geometric.utils.is_undirected' imported but unused
- 3 from torch_geometric.utils import degree, is_undirected
1 F821: undefined name 'aggr_index'
- 147 message = torch.sparse.sum(msgs, aggr_index).to_dense()
17 E501: line too long (99 > 79 characters) (and 11 similar)
- Line too long (99 > 79 characters):
  45 # different GSN variants: extended gin and general formulation (see supplementary material)
- Line too long (92 > 79 characters):
  50 self.central_node_id_encoder = central_encoder(kwargs['id_embedding'], d_id,
- Line too long (92 > 79 characters):
  51 extend=kwargs['extend_dims'])
- Line too long (89 > 79 characters):
  68 msg_input_dim = 2 * d_in + d_id if id_scope == 'local' else 2 * (d_in + d_id)
- Line too long (97 > 79 characters):
  81 raise NotImplementedError('msg kind {} is not currently supported.'.format(msg_kind))
- Line too long (80 > 79 characters):
  100 x = torch.cat([x, degrees], -1) if self.retain_features else degrees
- Line too long (96 > 79 characters):
  107 identifiers_ii, identifiers = self.central_node_id_encoder(identifiers, n_nodes)
- Line too long (101 > 79 characters):
  111 self.propagate(edge_index=edge_index, x=x, identifiers=identifiers))
- Line too long (104 > 79 characters):
  115 torch.cat((x, self.propagate(edge_index=edge_index, x=x, identifiers=identifiers)), -1))
- Line too long (106 > 79 characters):
  118 raise NotImplementedError("Message kind {} is not currently supported.".format(self.msg_kind))
- Line too long (85 > 79 characters):
  127 edge_index_i, edge_index_j = edge_index[select, :], edge_index[1 - select, :]
- Line too long (101 > 79 characters):
  135 identifiers_i, identifiers_j = identifiers[edge_index_i, :], identifiers[edge_index_j, :]
- Line too long (83 > 79 characters):
  138 msgs = self.message(x_i, x_j, identifiers_i, identifiers_j, identifiers_ij)
- Line too long (104 > 79 characters):
  139 msgs = torch.sparse.FloatTensor(edge_index, msgs, torch.Size([n_nodes, n_nodes, msgs.shape[1]]))
- Line too long (106 > 79 characters):
  151 raise NotImplementedError("Aggregation kind {} is not currently supported.".format(self.aggr))
- Line too long (106 > 79 characters):
  171 raise NotImplementedError("Message kind {} is not currently supported.".format(self.msg_kind))
- Line too long (109 > 79 characters):
  176 return '{}(msg_fn = {}, update_fn = {})'.format(self.__class__.__name__, self.msg_fn, self.update_fn)