Fix another PDB parsing bug

aqlaboratory · Jun 22, 2022 · 6da2cda · 6da2cda
1 parent 0f54f67
commit 6da2cda
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/openfold/np/protein.py b/openfold/np/protein.py
@@ -252,9 +252,9 @@ def add_pdb_headers(prot: Protein, pdb_str: str) -> str:
         out_pdb_lines.append(f"REMARK {remark}")
 
     parents_per_chain = None
-    if(prot.parents is not None):
+    if(prot.parents is not None and len(prot.parents) > 0):
         parents_per_chain = []
-        if(prot.parents_chain_index is not None and len(prot.parents_chain_index) > 0):
+        if(prot.parents_chain_index is not None):
             cur_chain = prot.parents_chain_index[0]
             parent_dict = {}
             for p, i in zip(prot.parents, prot.parents_chain_index):