Update asym chain names

chaidiscovery · Dec 24, 2024 · 5d16a1c · 5d16a1c
1 parent 4097a06
commit 5d16a1c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chai_lab/chai1.py b/chai_lab/chai1.py
@@ -906,9 +906,9 @@ def avg_per_token_1d(x):
             bfactors=scaled_plddt_scores_per_atom,
             output_batch=inputs,
             write_path=cif_out_path,
+            # Set asym names to be A, B, C, ...
             asym_entity_names={
-                i: c.entity_data.entity_name
-                for i, c in enumerate(feature_context.chains, start=1)
+                i + 1: chr(i + 65) for i in range(len(feature_context.chains))
             },
         )
         cif_paths.append(cif_out_path)