fix(data): match input format in multimer prediction

Return as dictionary.

Closes baker-laboratory/RoseTTAFold-All-Atom#11 (supersedes).
Fixes baker-laboratory/RoseTTAFold-All-Atom#12.
This commit is contained in:
Nuri Jung 2024-03-18 16:31:48 +09:00
parent 156e25ee49
commit bd57e10676
No known key found for this signature in database
GPG key ID: D5BDFF67E90BACBE

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:] ins[N:, start:start+L] = ins_orig[1:]
start += L start += L
return msa, ins return {"msa": msa, "ins": ins}
def merge_msas(a3m_list, L_s): def merge_msas(a3m_list, L_s):
""" """