From c6091e208160fec33f42954f4d1ded0cdcb91ae4 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 31 Jan 2026 05:05:09 +0000
Subject: [PATCH 1/2] Initial plan


From 83979b4a0e8377c5c7ff80a3eee8d1aec56fa9aa Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 31 Jan 2026 05:10:01 +0000
Subject: [PATCH 2/2] Fix IndexError for sequences longer than max_length in
 kaggle_submission.py

Co-authored-by: drqsatoshi <240532885+drqsatoshi@users.noreply.github.com>
---
 kaggle_submission.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kaggle_submission.py b/kaggle_submission.py
index ecacf709..58e52ca9 100644
--- a/kaggle_submission.py
+++ b/kaggle_submission.py
@@ -150,9 +150,9 @@ def predict_structures(
             
             # Store predictions
             for j, (idx, row) in enumerate(batch_df.iterrows()):
-                seq_len = len(row['sequence'])
+                seq_len = min(len(row['sequence']), max_length)
                 target_id = row['target_id']
-                sequence = row['sequence']
+                sequence = row['sequence'][:max_length]  # Truncate sequence to max_length
                 coords = pred_coords[j, :seq_len, :, :]
                 
                 predictions.append({