diff --git a/esm/sdk/api.py b/esm/sdk/api.py index 98ccea1..004d842 100644 --- 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