Fix PDB parsing bug

aqlaboratory · Jun 22, 2022 · 0f54f67 · 0f54f67
1 parent 99cdb06
commit 0f54f67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openfold/np/protein.py b/openfold/np/protein.py
@@ -254,7 +254,7 @@ def add_pdb_headers(prot: Protein, pdb_str: str) -> str:
     parents_per_chain = None
     if(prot.parents is not None):
         parents_per_chain = []
-        if(prot.parents_chain_index is not None):
+        if(prot.parents_chain_index is not None and len(prot.parents_chain_index) > 0):
             cur_chain = prot.parents_chain_index[0]
             parent_dict = {}
             for p, i in zip(prot.parents, prot.parents_chain_index):