Skip to content

Commit 36555f5

Browse files
committed
restrict ensemble testing to exclude Count targets, as unsupported
1 parent 5929f2f commit 36555f5

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/test.jl

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,8 @@ function next!(p)
99
MLJ.ProgressMeter.updateProgress!(p)
1010
end
1111

12+
const ENSEMBLE_TARGET_ELSCITYPE = Union{Missing, Continuous, Finite}
13+
1214
"""
1315
test(models, data...; mod=Main, level=2, throw=false, verbosity=1)
1416
@@ -382,15 +384,16 @@ function test(model_proxies, data...; mod=Main, level=2, throw=false, verbosity=
382384
outcome == "×" && continue
383385

384386
#[ensemble_prediction]:
385-
ensemble_prediction, outcome =
386-
MLJTestIntegration.ensemble_prediction(
387+
if target_scitype(model_type) <: AbstractVector{<:ENSEMBLE_TARGET_ELSCITYPE}
388+
ensemble_prediction, outcome = MLJTestIntegration.ensemble_prediction(
387389
model_instance,
388390
data...;
389391
throw,
390392
verbosity,
391393
)
392-
row = update(row, i, :ensemble_prediction, ensemble_prediction, outcome)
393-
outcome == "×" && continue
394+
row = update(row, i, :ensemble_prediction, ensemble_prediction, outcome)
395+
outcome == "×" && continue
396+
end
394397

395398
# [iteration_prediction]:
396399
if !isnothing(iteration_parameter(model_instance))

0 commit comments

Comments
 (0)