Skip to content

Commit 55af3fc

Browse files
authored
Update asym chain names (#262)
1 parent 4097a06 commit 55af3fc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

chai_lab/chai1.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -906,9 +906,9 @@ def avg_per_token_1d(x):
906906
bfactors=scaled_plddt_scores_per_atom,
907907
output_batch=inputs,
908908
write_path=cif_out_path,
909+
# Set asym names to be A, B, C, ...
909910
asym_entity_names={
910-
i: c.entity_data.entity_name
911-
for i, c in enumerate(feature_context.chains, start=1)
911+
i + 1: chr(i + 65) for i in range(len(feature_context.chains))
912912
},
913913
)
914914
cif_paths.append(cif_out_path)

0 commit comments

Comments
 (0)