From fb78132452b9f416f4997e5fbf1ecc18e8d9d8a8 Mon Sep 17 00:00:00 2001
From: Alex Morehead <alex.morehead@gmail.com>
Date: Mon, 11 Mar 2024 10:52:06 -0500
Subject: [PATCH] Fix merge_inputs.py

---
 rf2aa/data/merge_inputs.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rf2aa/data/merge_inputs.py b/rf2aa/data/merge_inputs.py
index bf12ff8..72aff8c 100644
--- a/rf2aa/data/merge_inputs.py
+++ b/rf2aa/data/merge_inputs.py
@@ -73,8 +73,8 @@ def merge_protein_inputs(protein_inputs, deterministic: bool = False):
         chain_lengths = list(zip(protein_inputs.keys(), lengths_list))
 
         merged_input = RawInputData(
-            a3m_out["msa"],
-            a3m_out["ins"],
+            a3m_out[0],
+            a3m_out[1],
             bond_feats,
             xyz_t[:max_template_dim],
             mask_t[:max_template_dim],