(protein, chis)
| 721 | |
| 722 | @torch.no_grad() |
| 723 | def set_chis(protein, chis): |
| 724 | assert chis.shape[0] == protein.num_residue |
| 725 | assert chis.shape[1] == 4 |
| 726 | cur_chis = get_chis(protein) |
| 727 | chi_to_rotate = chis - cur_chis |
| 728 | chi_to_rotate[torch.isnan(chi_to_rotate)] = 0 |
| 729 | rotate_side_chain(protein, chi_to_rotate) |
| 730 | return protein |
| 731 | |
| 732 | |
| 733 | @torch.no_grad() |
nothing calls this directly
no test coverage detected