mlr3 - predict.all = TRUE のレンジャー学習器のバグ

Question

パッケージ {ranger} のランダム フォレストのすべての予測を ml3 予測オブジェクトに保存し、個々のツリーの予測を別の学習者の特徴として使用したいと考えています。

次のコードは、R の次のエラー メッセージに続きます。

コード：

library("mlr3")
library("mlr3learners")

task = tsk("iris")
learner = lrn("classif.ranger", predict.all = TRUE)

# Train
train_set = sample(task$nrow, 0.8 * task$nrow)
test_set = setdiff(seq_len(task$nrow), train_set)

learner$train(task, row_ids = train_set)

# Predition
prediction = learner$predict(task, row_ids = test_set)
print(prediction)

エラー：

check_prediction_data.PredictionDataClassif(pdata) のエラー:
'as_factor(pdata$response, levels = lvls)' のアサーションが失敗しました: 長さは 30 でなければなりませんが、長さは 15000 です。

誰かがこの問題を解決するのを手伝ってくれますか?