github-mirrors/RoseTTAFold-All-Atom
1
0
Fork 0
mirror of https://github.com/baker-laboratory/RoseTTAFold-All-Atom.git synced 2024-11-24 22:37:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #49 from seoklab/backport

Browse source 
Fix some bugs in data handling code
This commit is contained in:
Rohith Krishna 2024-03-23 16:29:26 -07:00 committed by GitHub
commit b461e12016
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rf2aa/data/data_loader_utils.py
View file

@ -455,7 +455,7 @@ def merge_a3m_homo(msa_orig, ins_orig, nmer, mode="default"):
ins[N:, start:start+L] = ins_orig[1:]
start += L
return msa, ins
return {"msa": msa, "ins": ins}
def merge_msas(a3m_list, L_s):
"""

4
rf2aa/data/merge_inputs.py
View file

@ -43,8 +43,8 @@ def merge_protein_inputs(protein_inputs, deterministic: bool = False):
if len(unique_a3m) >1:
a3m_out = unique_a3m[0]
for i in range(1, len(unique_a3m)):
a3m_out = join_msas_by_taxid(a3m_out, a3m_list[i])
for unq_a3m in unique_a3m[1:]:
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)
else:
a3m = unique_a3m[0]