minor bug fix in (Xte/Yte transpose)

amarquand · Oct 12, 2020 · e0440ec · e0440ec
1 parent 4d9fd55
commit e0440ec
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pcntoolkit/normative.py b/pcntoolkit/normative.py
@@ -300,11 +300,11 @@ def estimate(covfile, respfile, **kwargs):
         cvfolds = 1
         Xte = fileio.load(testcov)
         if len(Xte.shape) == 1:
-            Xte = Xte[:, np.newaxis].T
+            Xte = Xte[:, np.newaxis]
         if testresp is not None:
             Yte, testmask = load_response_vars(testresp, maskfile)
             if len(Yte.shape) == 1:
-                Yte = Yte[:, np.newaxis].T
+                Yte = Yte[:, np.newaxis]
         else:
             sub_te = Xte.shape[0]
             Yte = np.zeros([sub_te, Nmod])