(self, graph, surf_graph)
| 95 | raise ValueError("Unknown readout `%s`" % readout) |
| 96 | |
| 97 | def residue2surface(self, graph, surf_graph): |
| 98 | res2surf = graph.res2surf.flatten(1, 2) |
| 99 | # Re-index after batching surface graphs |
| 100 | res2surf = res2surf + (surf_graph.num_cum_nodes - surf_graph.num_nodes)[graph.residue2graph].unsqueeze(-1) |
| 101 | |
| 102 | def surface_feature_init(self, graph, surf_graph, input): |
| 103 | # Residue -> surface graph correspondence |