Fix merge_inputs.py

This commit is contained in:
Alex Morehead 2024-03-11 10:52:06 -05:00 committed by GitHub
parent e51736412b
commit fb78132452
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -73,8 +73,8 @@ def merge_protein_inputs(protein_inputs, deterministic: bool = False):
chain_lengths = list(zip(protein_inputs.keys(), lengths_list)) chain_lengths = list(zip(protein_inputs.keys(), lengths_list))
merged_input = RawInputData( merged_input = RawInputData(
a3m_out["msa"], a3m_out[0],
a3m_out["ins"], a3m_out[1],
bond_feats, bond_feats,
xyz_t[:max_template_dim], xyz_t[:max_template_dim],
mask_t[:max_template_dim], mask_t[:max_template_dim],