Merge pull request #67 from alan-turing-institute/scitype1

Bump [compat] ScientificTypes = "^1"

Author: ablaom

Project.toml

@@ -1,14 +1,14 @@
 name = "MLJModelInterface"
 uuid = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
 authors = ["Thibaut Lienart and Anthony Blaom"]
-version = "0.3.4"
+version = "0.3.5"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 ScientificTypes = "321657f4-b219-11e9-178b-2701a2544e81"
 
 [compat]
-ScientificTypes = "^0.7,^0.8"
+ScientificTypes = "^1"
 julia = "1"
 
 [extras]

test/model_api.jl

@@ -25,3 +25,44 @@ mutable struct APIx1 <: Static end
     #update fallback = fit
     @test update(m0, 1, 5, nothing, randn(2), 5) == (5, nothing, nothing)
 end
+
+struct DummyUnivariateFinite end
+
+mutable struct UnivariateFiniteFitter <: Probabilistic end
+
+@testset "models fitting a distribution to data" begin
+
+    function MLJModelInterface.fit(model::UnivariateFiniteFitter,
+                                   verbosity::Int, X, y)
+
+        fitresult = DummyUnivariateFinite()
+        report = nothing
+        cache = nothing
+
+        verbosity > 0 && @info "Fitted a $fitresult"
+
+        return fitresult, cache, report
+    end
+
+    MLJModelInterface.predict(model::UnivariateFiniteFitter,
+                          fitresult,
+                          X) = fill(fitresult, length(X))
+
+    MLJModelInterface.input_scitype(::Type{<:UnivariateFiniteFitter}) =
+        AbstractVector{Nothing}
+    MLJModelInterface.target_scitype(::Type{<:UnivariateFiniteFitter}) =
+        AbstractVector{<:Finite}
+
+    y =categorical(collect("aabbccaa"))
+    X = fill(nothing, length(y))
+    model = UnivariateFiniteFitter()
+    fitresult, cache, report = MLJModelInterface.fit(model, 1, X, y)
+
+    @test cache == nothing
+    @test report == nothing
+
+    ytest = y[1:3]
+    yhat = predict(model, fitresult, fill(nothing, 3))
+    @test yhat == fill(DummyUnivariateFinite(), 3)
+
+end