From e9f060214d19a6420a69cd1187fdaf81978791fc Mon Sep 17 00:00:00 2001
From: carolynkim <carolynkim@users.noreply.github.com>
Date: Thu, 27 Jun 2024 16:39:54 -0700
Subject: [PATCH] fix: save plddt to pdb (#14)

---
 esm/sdk/api.py | 1 +
 1 file changed, 1 insertion(+)

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