Skip to content

Commit 06eabf3

Browse files
authored
fix: assert qauntiles in ranger (#355)
1 parent 97c1961 commit 06eabf3

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

DESCRIPTION

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ URL: https://mlr3learners.mlr-org.com,
2626
https://github.com/mlr-org/mlr3learners
2727
BugReports: https://github.com/mlr-org/mlr3learners/issues
2828
Depends:
29-
mlr3 (>= 0.23.0),
29+
mlr3 (>= 1.1.0.9000),
3030
R (>= 3.3.0)
3131
Imports:
3232
checkmate,
@@ -50,6 +50,8 @@ Suggests:
5050
rmarkdown,
5151
testthat (>= 3.0.0),
5252
xgboost (>= 1.6.0)
53+
Remotes:
54+
mlr-org/mlr3
5355
Config/testthat/edition: 3
5456
Encoding: UTF-8
5557
NeedsCompilation: no

R/LearnerRegrRanger.R

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,7 @@ LearnerRegrRanger = R6Class("LearnerRegrRanger",
141141
.args = pv)
142142

143143
if (self$predict_type == "quantiles") {
144+
assert_quantiles(self, quantile_response = TRUE)
144145
quantiles = prediction$predictions
145146
setattr(quantiles, "probs", private$.quantiles)
146147
setattr(quantiles, "response", private$.quantile_response)

0 commit comments

Comments
 (0)