Skip to content

Commit a74110b

Browse files
committed
make sure :transform is tested for transformers
1 parent 3ea48e1 commit a74110b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/attemptors.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,9 @@ function operations(fitted_machine, data...; throw=false, verbosity=1)
9393
model = fitted_machine.model
9494
operations = String[]
9595
methods = MLJBase.implemented_methods(fitted_machine.model)
96+
if model isa Static && !(:transform in methods)
97+
push!(methods, :transform)
98+
end
9699
_, test = MLJBase.partition(1:MLJBase.nrows(first(data)), 0.5)
97100
if :predict in methods
98101
predict(fitted_machine, first(data))

0 commit comments

Comments
 (0)