diff --git a/Project.toml b/Project.toml index 1533d284..8f50c00d 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "MLJBase" uuid = "a7f614a8-145f-11e9-1d2a-a57a1082229d" authors = ["Anthony D. Blaom "] -version = "0.21.12" +version = "0.21.13" [deps] CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597" @@ -36,7 +36,7 @@ ComputationalResources = "0.3" DelimitedFiles = "1" Distributions = "0.25.3" InvertedIndices = "1" -LossFunctions = "0.10" +LossFunctions = "0.11" MLJModelInterface = "1.7" Missings = "0.4, 1" OrderedCollections = "1.1" diff --git a/src/MLJBase.jl b/src/MLJBase.jl index b3d89bd3..63fe4fd7 100644 --- a/src/MLJBase.jl +++ b/src/MLJBase.jl @@ -79,6 +79,7 @@ import .Threads # Operations & extensions import LossFunctions +import LossFunctions.Traits import StatsBase import StatsBase: fit!, mode, countmap import Missings: levels