hot fix

Author: tlienart

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJLinearModels"
 uuid = "6ee0df7b-362f-4a72-a706-9e79364fb692"
 authors = ["Thibaut Lienart <[email protected]>"]
-version = "0.3.5"
+version = "0.3.6"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"

src/mlj/interface.jl

@@ -45,7 +45,8 @@ function MMI.fit(m::Union{CLF_MODELS...}, verb::Int, X, y)
     sch = MMI.schema(X)
     features = (sch === nothing) ? nothing : sch.names
     yplain   = convert.(Int, MMI.int(y))
-    classes  = MMI.classes(y[1])[unique(yplain)]
+    decoder  = MMI.decoder(y[1])
+    classes  = decoder(sort(unique(yplain)))
     nclasses = length(classes)
     if nclasses == 2
         # recode

test/fit/robust.jl

@@ -96,7 +96,7 @@ end
     @test isapprox(J(θ_iwls),     0.818677, rtol=1e-4)
 end
 
-@testset "LogisticRReg" begin
+@testset "LogisticReg" begin
     δ = 1.5
     λ = 1.0
     rr = RobustRegression(rho=Logistic(delta=δ), lambda=λ, penalize_intercept=true)

test/runtests.jl

@@ -1,7 +1,7 @@
 using MLJLinearModels, Test, LinearAlgebra, Random, StableRNGs
 import MLJBase # not MLJModelInterface, to mimic the full interface
 
-DO_COMPARISONS = false; include("testutils.jl")
+DO_COMPARISONS = true; include("testutils.jl")
 
 m("UTILS"); include("utils.jl")