Update test

eliascarv · eliascarv · commit e7f945d84630 · 2023-09-27T14:33:30.000-03:00
diff --git a/ext/StatsLearnModelsMLJModelInterfaceExt.jl b/ext/StatsLearnModelsMLJModelInterfaceExt.jl
@@ -26,7 +26,7 @@ function SLM.predict(fmodel::SLM.FittedModel{<:MI.Model}, table)
   else
     MI.predict(model, fitresult, data...)
   end
-  (; target => ŷ)
+  (; target => ŷ) |> Tables.materializer(table)
 end
 
 end
diff --git a/test/Project.toml b/test/Project.toml
@@ -2,4 +2,5 @@
 DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
 MLJ = "add582a8-e3ab-11e8-2d5e-e98b27df1bc7"
 MLJDecisionTreeInterface = "c6f25543-311c-4c74-83dc-3ea6d1015661"
+Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1,9 +1,11 @@
 import StatsLearnModels as SLM
 using MLJ, MLJDecisionTreeInterface
 using DataFrames
+using Random
 using Test
 
 @testset "StatsLearnModels.jl" begin
+  Random.seed!(123)
   iris = DataFrame(load_iris())
   input = iris[:, Not(:target)]
   output = iris[:, [:target]]
