def METHOD_NAME(self, x, edge_index, edge_attr):
    x = self.node_encoder(x)
    edge_attr = self.edge_encoder(edge_attr)
    x = self.layers[0].conv(x, edge_index, edge_attr)
    for layer in self.layers[1:]:
        x = layer(x, edge_index, edge_attr)
    x = self.layers[0].act(self.layers[0].norm(x))
    x = F.dropout(x, p=0.1, training=self.training)
    return self.lin(x)