Skip to content

Commit 7bd7c68

Browse files
authoredAug 25, 2024
Merge pull request #193 from zhi-yi-huang/issue#186
fix: fix issue#186
2 parents ecc66e3 + 9c2a8e8 commit 7bd7c68

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed
 

‎causallearn/score/LocalScoreFunction.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def local_score_cv_general(
198198
PAi = list(PAi)
199199

200200
T = Data.shape[0]
201-
X = Data[:, Xi]
201+
X = Data[:, [Xi]]
202202
var_lambda = parameters["lambda"] # regularization parameter
203203
k = parameters["kfold"] # k-fold cross validation
204204
n0 = math.floor(T / k)
@@ -715,7 +715,7 @@ def local_score_marginal_general(
715715
"""
716716

717717
T = Data.shape[0]
718-
X = Data[:, Xi]
718+
X = Data[:, [Xi]]
719719
dX = X.shape[1]
720720

721721
# set the kernel for X

0 commit comments

Comments
 (0)