From e523b21f059127262cc757582ff2128e80157af8 Mon Sep 17 00:00:00 2001
From: JaiDhyani <jaiwithani@gmail.com>
Date: Wed, 6 Mar 2024 17:26:47 -0800
Subject: [PATCH] asdf

---
 src/delphi/train/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/delphi/train/training.py b/src/delphi/train/training.py
index d9e6c79c..dc1e876b 100644
--- a/src/delphi/train/training.py
+++ b/src/delphi/train/training.py
@@ -194,7 +194,7 @@ def train_step(
             callback(eval_data)
 
     if iter_num == 0 and config.eval_only:
-        return True, None, None, None
+        return True, None, None, None, None
 
     # 3. forward backward update, with optional gradient accumulation to simulate larger batch size
     X, Y = next(train_batch_iter)