Skip to content

GitLab

"sgl-kernel/vscode:/vscode.git/clone" did not exist on "a55cb8b2f86cf5c28a4972f3df765dbd77e19947"
Commit 91b43223 authored by Rich Evans's avatar Rich Evans Committed by Copybara-Service
Browse files

Fix PAE visualization for single chain predictions. 

PiperOrigin-RevId: 407773747
Change-Id: I8635dd77ecf366f4f5a362993b964adffc3ff186
parent c128d1aa
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
notebooks/AlphaFold.ipynb
View file @ 91b43223
...@@ -648,6 +648,7 @@ ...@@ -648,6 +648,7 @@
" total_num_res = best_unrelaxed_prot.residue_index.shape[-1]\n", " total_num_res = best_unrelaxed_prot.residue_index.shape[-1]\n",
" chain_ids = best_unrelaxed_prot.chain_index\n", " chain_ids = best_unrelaxed_prot.chain_index\n",
" for chain_boundary in np.nonzero(chain_ids[:-1] - chain_ids[1:]):\n", " for chain_boundary in np.nonzero(chain_ids[:-1] - chain_ids[1:]):\n",
" if chain_boundary.size:\n",
" plt.plot([0, total_num_res], [chain_boundary, chain_boundary], color='red')\n", " plt.plot([0, total_num_res], [chain_boundary, chain_boundary], color='red')\n",
" plt.plot([chain_boundary, chain_boundary], [0, total_num_res], color='red')\n", " plt.plot([chain_boundary, chain_boundary], [0, total_num_res], color='red')\n",
"\n", "\n",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment