Extend compat for MLJModelInterface and update tests to reflect MLJBase changes (#88)

* Update compatibility MLJModelInterface = "^0.3,^0.4"; bump to 0.5.2

* update test to reflect migration of info_dict from MLJBase

Author: ablaom

Project.toml

@@ -1,7 +1,7 @@
 name = "MLJLinearModels"
 uuid = "6ee0df7b-362f-4a72-a706-9e79364fb692"
 authors = ["Thibaut Lienart <[email protected]>"]
-version = "0.5.1"
+version = "0.5.2"
 
 [deps]
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
@@ -16,8 +16,8 @@ Parameters = "d96e819e-fc66-5662-9728-84c9c7592b0a"
 DocStringExtensions = "^0.8"
 IterativeSolvers = "^0.8"
 LinearMaps = "^2.6"
-MLJModelInterface = "^0.3"
-Optim = "0.20,0.21,1,1.1"
+MLJModelInterface = "^0.3,^0.4"
+Optim = "0.20,0.21,1"
 Parameters = "^0.12"
 julia = "^1"
 

test/interface/meta.jl

@@ -1,29 +1,32 @@
+import MLJBase: package_name, package_url, package_license,
+    is_wrapper, is_pure_julia, input_scitype, target_scitype,
+    supports_weights, docstring, load_path
+
 @testset "meta-pkg" begin
+    @show methods(package_name)
     rr = RidgeRegressor()
-    d = MLJBase.info_dict(rr)
-    @test d[:package_name] == "MLJLinearModels"
-    @test d[:package_url] == "https://github.com/alan-turing-institute/MLJLinearModels.jl"
-    @test d[:package_license] == "MIT"
-    @test d[:is_pure_julia] == true
-    @test d[:is_wrapper] == false
+    @test package_name(rr) == "MLJLinearModels"
+    @test package_url(rr) ==
+        "https://github.com/alan-turing-institute/MLJLinearModels.jl"
+    @test package_license(rr) == "MIT"
+    @test is_pure_julia(rr) == true
+    @test is_wrapper(rr) == false
 end
 
 @testset "meta-reg" begin
     lr = LinearRegressor()
-    d = MLJBase.info_dict(lr)
-    @test d[:input_scitype] == MLJBase.Table(MLJBase.Continuous)
-    @test d[:target_scitype] == AbstractVector{MLJBase.Continuous}
-    @test d[:supports_weights] == false
-    @test !isempty(d[:docstring])
-    @test d[:load_path] == "MLJLinearModels.LinearRegressor"
+    @test input_scitype(lr) == MLJBase.Table(MLJBase.Continuous)
+    @test target_scitype(lr) == AbstractVector{MLJBase.Continuous}
+    @test supports_weights(lr) == false
+    @test !isempty(docstring(lr))
+    @test load_path(lr) == "MLJLinearModels.LinearRegressor"
 end
 
 @testset "meta-clf" begin
     lr = LogisticClassifier()
-    d = MLJBase.info_dict(lr)
-    @test d[:input_scitype] == MLJBase.Table(MLJBase.Continuous)
-    @test d[:target_scitype] == AbstractVector{<:MLJBase.Finite}
-    @test d[:supports_weights] == false
-    @test !isempty(d[:docstring])
-    @test d[:load_path] == "MLJLinearModels.LogisticClassifier"
+    @test input_scitype(lr) == MLJBase.Table(MLJBase.Continuous)
+    @test target_scitype(lr) == AbstractVector{<:MLJBase.Finite}
+    @test supports_weights(lr) == false
+    @test !isempty(docstring(lr))
+    @test load_path(lr) == "MLJLinearModels.LogisticClassifier"
 end