fix(data): join unique a3m results only

Fixes seoklab/RoseTTAFold-All-Atom#3.
This commit is contained in:
Nuri Jung 2024-03-18 16:21:28 +09:00
parent c09141f1d5
commit 156e25ee49
No known key found for this signature in database
GPG key ID: D5BDFF67E90BACBE

View file

@ -43,8 +43,8 @@ def merge_protein_inputs(protein_inputs, deterministic: bool = False):
if len(unique_a3m) >1: if len(unique_a3m) >1:
a3m_out = unique_a3m[0] a3m_out = unique_a3m[0]
for i in range(1, len(unique_a3m)): for unq_a3m in unique_a3m[1:]:
a3m_out = join_msas_by_taxid(a3m_out, a3m_list[i]) a3m_out = join_msas_by_taxid(a3m_out, unq_a3m)
a3m_out = expand_multi_msa(a3m_out, unique_hashes, hash_list, unique_lengths_list, lengths_list) a3m_out = expand_multi_msa(a3m_out, unique_hashes, hash_list, unique_lengths_list, lengths_list)
else: else:
a3m = unique_a3m[0] a3m = unique_a3m[0]