From 10077d8a8e120f632dee0ea25e68008c4993b535 Mon Sep 17 00:00:00 2001 From: Zeming Lin Date: Tue, 6 Aug 2024 16:17:44 +0000 Subject: [PATCH] Fix examples/raw_forward.py --- examples/raw_forwards.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/raw_forwards.py b/examples/raw_forwards.py index 6651d78..af41dbb 100644 --- a/examples/raw_forwards.py +++ b/examples/raw_forwards.py @@ -26,7 +26,7 @@ def inverse_folding_example(): encoder = ESM3_structure_encoder_v0("cuda") model = ESM3_sm_open_v0("cuda") - chain = ProteinChain.from_pdb("esm/data/1utn.pdb") + chain = ProteinChain.from_rcsb("1utn", "A") coords, plddt, residue_index = chain.to_structure_encoder_inputs() coords = coords.cuda() plddt = plddt.cuda()