Skip to content

Commit 7caaa9b

Browse files
committed
fix fitted_machine attemptor for Static models
1 parent 4c6aa85 commit 7caaa9b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/attemptors.jl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,8 @@ end
8989
function fitted_machine(model, data...; throw=false, verbosity=1)
9090
message = "[:fitted_machine] Fitting machine "
9191
attempt(finalize(message, verbosity); throw) do
92-
mach = machine(model, data...)
92+
mach = model isa Static ? machine(model) :
93+
machine(model, data...)
9394
fit!(mach, verbosity=-1)
9495
MLJ.report(mach)
9596
MLJ.fitted_params(mach)

0 commit comments

Comments
 (0)