updating MLJBase bound and tests (#42)

* updating MLJBase bound and tests

Author: tlienart

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJLinearModels"
 uuid = "6ee0df7b-362f-4a72-a706-9e79364fb692"
 authors = ["Thibaut Lienart <[email protected]>"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
@@ -17,7 +17,7 @@ Tables = "bd369af6-aec1-5ad0-b16a-f7cc5008161c"
 DocStringExtensions = "^0.8"
 IterativeSolvers = "^0.8"
 LinearMaps = "^2.5"
-MLJBase = "^0.8"
+MLJBase = "^0.9"
 Optim = "^0.19"
 Parameters = "^0.12"
 Tables = "^0.2"

test/glr/grad-hess-prox.jl

@@ -106,7 +106,7 @@ end
     g1 = similar(θ1)
     H1 = zeros(p+1, p+1)
     f1 = fgh!(f1, g1, H1, θ1)
-    @test f1 == J(θ1)
+    @test f1 ≈ J(θ1)
     @test g1 ≈              -X1' * (y .* R.σ.(-y .* (X1 * θ1))) .+ λ .* θ1
     @test H1 ≈               X1' * (Diagonal(R.σ.(y .* (X1 * θ1))) * X1) + λ * I
 
@@ -127,7 +127,7 @@ end
     g1 = similar(θ1)
     H1 = zeros(p+1, p+1)
     f1 = fgh!(f1, g1, H1, θ1)
-    @test f1 == J(θ1)
+    @test f1 ≈ J(θ1)
     @test g1 ≈              -X1' * (y .* R.σ.(-y .* (X1 * θ1))) .+ λ .* θ1 .* maskint
     @test H1 ≈               X1' * (Diagonal(R.σ.(y .* (X1 * θ1))) * X1) + λ * Diagonal(maskint)
     Hv! = R.Hv!(lr1, X, y)
@@ -243,7 +243,7 @@ end
     f1 = fgh1!(0.0, g1, H1, θ1_)
     r1 = X1*θ1_ .- y1
     mask = abs.(r1) .<= δ
-    @test f1 == hlr1.loss(r1) + hlr1.penalty(θ1_)
+    @test f1 ≈ hlr1.loss(r1) + hlr1.penalty(θ1_)
     @test g1 ≈              (X1' * (r1 .* mask)) .+ (X1' * (δ .* sign.(r1) .* .!mask)) .+ λ .* θ1_
     @test H1 ≈               X1' * (mask .* X1) + λ*I
 
@@ -266,7 +266,7 @@ end
     f1 = fgh1!(0.0, g1, H1, θ1_)
     r1 = X1*θ1_ .- y1
     mask = abs.(r1) .<= δ
-    @test f1 == hlr1.loss(r1) + hlr1.penalty(θ1_ .* maskint)
+    @test f1 ≈ hlr1.loss(r1) + hlr1.penalty(θ1_ .* maskint)
     @test g1 ≈              (X1' * (r1 .* mask)) .+ (X1' * (δ .* sign.(r1) .* .!mask)) .+
                                 λ .* θ1_ .* maskint
     @test H1 ≈               X1' * (mask .* X1) + λ * Diagonal(maskint)