(protein, node_position=None)
| 672 | |
| 673 | |
| 674 | def get_chis(protein, node_position=None): |
| 675 | if node_position is None: |
| 676 | node_position = protein.node_position |
| 677 | chi_atom_position = get_chi_atom_position(protein, node_position) # (num_residue, 4, 4, 3) |
| 678 | chis = get_dihedral(*chi_atom_position.unbind(-2)) # (num_residue, 4) |
| 679 | chis[~protein.chi_mask] = np.nan |
| 680 | return chis |
| 681 | |
| 682 | |
| 683 | def get_chi_atom_position(protein, node_position=None): |
no test coverage detected