Skip to content

Commit 5c041a5

Browse files
committed
make y_output optional in sub-function for sake of testing
1 parent 6ee5fcf commit 5c041a5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

forestci/forestci.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def _bias_correction(V_IJ, inbag, pred_centered, n_trees):
200200
return V_IJ_unbiased
201201

202202

203-
def _centered_prediction_forest(forest, X_test, y_output):
203+
def _centered_prediction_forest(forest, X_test, y_output=None):
204204
"""
205205
Center the tree predictions by the mean prediction (forest)
206206

0 commit comments

Comments
 (0)