@@ -727,7 +727,7 @@ set_fit(
727727 value = list (
728728 interface = " data.frame" ,
729729 data = c(x = " X" , y = " Y" , weights = " sample.weights" ),
730- protect = c(" x " , " y " , " weights" ),
730+ protect = c(" X " , " Y " , " weights" ),
731731 func = c(pkg = " grf" , fun = " probability_forest" ),
732732 defaults = list (
733733 num.threads = 1
@@ -753,8 +753,8 @@ set_fit(
753753 mode = " regression" ,
754754 value = list (
755755 interface = " data.frame" ,
756- data = c(x = " X" , y = " Y" , weights = " case .weights" ),
757- protect = c(" x " , " y " , " weights" ),
756+ data = c(x = " X" , y = " Y" , weights = " sample .weights" ),
757+ protect = c(" X " , " Y " , " weights" ),
758758 func = c(pkg = " grf" , fun = " regression_forest" ),
759759 defaults = list (
760760 num.threads = 1
@@ -860,8 +860,8 @@ set_fit(
860860 mode = " quantile regression" ,
861861 value = list (
862862 interface = " data.frame" ,
863- data = c(x = " X" , y = " Y" , weights = " case.weights " ),
864- protect = c(" x " , " y " , " weights " ),
863+ data = c(x = " X" , y = " Y" ),
864+ protect = c(" X " , " Y " ),
865865 func = c(pkg = " grf" , fun = " quantile_forest" ),
866866 defaults = list (
867867 num.threads = 1 ,
0 commit comments