From 1fcd8ec1a95a1e99ec16a560ef51a22f0b1a21a2 Mon Sep 17 00:00:00 2001
From: Alex Morehead <alex.morehead@gmail.com>
Date: Mon, 11 Mar 2024 19:01:21 -0500
Subject: [PATCH] Revert integer indexing in 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 72aff8c..bf12ff8 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[0],
-            a3m_out[1],
+            a3m_out["msa"],
+            a3m_out["ins"],
             bond_feats,
             xyz_t[:max_template_dim],
             mask_t[:max_template_dim],