From 63a23c2c886608c2ee4b65ea12ba2d330bc91a32 Mon Sep 17 00:00:00 2001
From: Kushagra  Srivastav <76401345+Kushagra7777@users.noreply.github.com>
Date: Thu, 3 Oct 2024 22:03:09 +0530
Subject: [PATCH] added a line to use test input as the first Q-state

---
 ARC_functions.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ARC_functions.py b/ARC_functions.py
index 57d1af4bc..487408af1 100644
--- a/ARC_functions.py
+++ b/ARC_functions.py
@@ -189,6 +189,8 @@ def ARC_main(tasks):
             # Convert the padded input array to binary format
             inp_binary = ARC_to_binary(inp_padded)
 
+            arcAgent.story[arcAgent.state - 1, arcAgent.arch.Q__flat] = inp_binary
+
             onp = np.asarray(pair['output'])  # Convert output data to NumPy array
             # Pad the output array
             onp_padded = pad_ARC(onp)