Merge pull request #193 from zhi-yi-huang/issue#186

fix: fix issue#186
py-why · Aug 25, 2024 · 7bd7c68 · 7bd7c68
2 parents ecc66e3 + 9c2a8e8
commit 7bd7c68
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/causallearn/score/LocalScoreFunction.py b/causallearn/score/LocalScoreFunction.py
@@ -198,7 +198,7 @@ def local_score_cv_general(
     PAi = list(PAi)
 
     T = Data.shape[0]
-    X = Data[:, Xi]
+    X = Data[:, [Xi]]
     var_lambda = parameters["lambda"]  # regularization parameter
     k = parameters["kfold"]  # k-fold cross validation
     n0 = math.floor(T / k)
@@ -715,7 +715,7 @@ def local_score_marginal_general(
     """
 
     T = Data.shape[0]
-    X = Data[:, Xi]
+    X = Data[:, [Xi]]
     dX = X.shape[1]
 
     # set the kernel for X