From 55af3fc95bcc8c90ebc3fd2e53d3e5a42d6fffe1 Mon Sep 17 00:00:00 2001
From: Kevin Wu <kevin@chaidiscovery.com>
Date: Tue, 24 Dec 2024 13:58:43 -0800
Subject: [PATCH] Update asym chain names (#262)

---
 chai_lab/chai1.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/chai_lab/chai1.py b/chai_lab/chai1.py
index 57c27fc..bb52c84 100644
--- 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)