From 23ddbd709779080744efe55b987b8a06d972eeef Mon Sep 17 00:00:00 2001
From: ParsaVahidi <pvahidi@usc.edu>
Date: Thu, 1 May 2025 17:51:27 -0700
Subject: [PATCH] Fixes minor bug in trial-based IPSID

---
 source/PSID/IPSID.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/source/PSID/IPSID.py b/source/PSID/IPSID.py
index 068286f..12559d1 100644
--- a/source/PSID/IPSID.py
+++ b/source/PSID/IPSID.py
@@ -720,7 +720,7 @@ def IPSID(
         w = Xk_Plus1[:n1, :] - XkP1Hat[:n1, :]  # Eq.(33)
     else:
         A = np.empty([0, 0])
-        w = np.empty([0, N])
+        w = np.empty([0, NTot])
 
     if n2 > 0:
         # A associated with the other states (X2)