From c21e1cced98bc58d7fddc8af49aa544a8133d215 Mon Sep 17 00:00:00 2001
From: Yi Wang <37149810+yiwang12@users.noreply.github.com>
Date: Tue, 12 Nov 2024 14:20:39 +0100
Subject: [PATCH] Update test_small_run.py

---
 tests/test_small_run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_small_run.py b/tests/test_small_run.py
index 92c41eb..c77da0b 100644
--- a/tests/test_small_run.py
+++ b/tests/test_small_run.py
@@ -35,7 +35,7 @@ def _run(
         random.seed(10)
         ninduced = round(list_D_chunked[ksample]["X"].shape[0] * 0.35)
         D_tmp = list_D_chunked[ksample]
-        D[ksample]["Z"] = D_tmp["X"][random.sample(range(0, D_tmp["X"].shape[0] - 1), ninduced), :]
+        D_tmp[ksample]["Z"] = D_tmp["X"][random.sample(range(0, D_tmp["X"].shape[0] - 1), ninduced), :]
 
     # step 1 initialize model
     fit = process_multiSample.ini_multiSample(list_D_chunked, n_loadings, "nb", chol=False)