res2surf is None

[9421g]

gvp.py里面residue2surface没有return，但是不会报错，是故意的还是不小心的。

# line 97
def residue2surface(self, graph, surf_graph):
        res2surf = graph.res2surf.flatten(1, 2)
        # Re-index after batching surface graphs
        res2surf = res2surf + (surf_graph.num_cum_nodes - surf_graph.num_nodes)[graph.residue2graph].unsqueeze(-1)  


# line 137
        # Surface -> residue graph correspondence
        res2surf = self.residue2surface(graph, surf_graph)

# line 151
surf_node_feature = surf_node_feature[res2surf].mean(dim=1)     # Mean readout from surface to residue graph
# res2surf is None
# surf_node_feature.shape [N, 256] -> [1, 256]