(self, sequence_graph, structure_graph)
| 100 | return sequence_graph, structure_graph, surface_graph |
| 101 | |
| 102 | def assign_structure(self, sequence_graph, structure_graph): |
| 103 | graph = structure_graph.clone() |
| 104 | # Assume the backbone structure won't change |
| 105 | assert graph.num_residue == sequence_graph.num_residue |
| 106 | with graph.residue(): |
| 107 | graph.residue_type = sequence_graph.residue_type |
| 108 | return graph |
| 109 | |
| 110 | def __getitem__(self, index): |
| 111 | sequence_graph = self.mutated_sequences[index] |