Update test_small_run.py

hansenlab · Nov 12, 2024 · c21e1cc · c21e1cc
1 parent b4d2ec4
commit c21e1cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)