fix: save plddt to pdb (evolutionaryscale#14)

jxshi · Jun 27, 2024 · e9f0602 · e9f0602
1 parent 41207fd
commit e9f0602
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/esm/sdk/api.py b/esm/sdk/api.py
@@ -101,6 +101,7 @@ def to_protein_chain(self) -> ProteinChain:
             entity_id=None,
             residue_index=None,
             insertion_code=None,
+            confidence=None if self.plddt is None else self.plddt.detach().cpu().numpy(),
         )
         return protein_chain