Merge pull request #15 from JuliaAI/fix-for-static

Fix for Static models

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJTestIntegration"
 uuid = "697918b4-fdc1-4f9e-8ff9-929724cee270"
 authors = ["Anthony D. Blaom <[email protected]>"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"

src/attemptors.jl

@@ -89,7 +89,8 @@ end
 function fitted_machine(model, data...; throw=false, verbosity=1)
     message = "[:fitted_machine] Fitting machine "
     attempt(finalize(message, verbosity); throw)  do
-        mach = machine(model, data...)
+        mach = model isa Static ? machine(model) :
+                                  machine(model, data...)
         fit!(mach, verbosity=-1)
         MLJ.report(mach)
         MLJ.fitted_params(mach)