Revert integer indexing in merge_inputs.py

This commit is contained in:
Alex Morehead 2024-03-11 19:01:21 -05:00 committed by GitHub
parent ec93ac6346
commit 1fcd8ec1a9
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[0], a3m_out["msa"],
a3m_out[1], a3m_out["ins"],
bond_feats, bond_feats,
xyz_t[:max_template_dim], xyz_t[:max_template_dim],
mask_t[:max_template_dim], mask_t[:max_template_dim],