diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index bc93431..39f6e20 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -18,6 +18,7 @@ jobs: fail-fast: false matrix: version: + - '1.9' - '1.10' os: - ubuntu-latest diff --git a/Project.toml b/Project.toml index 5f8e1e3..4499cf8 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ authors = ["jeremiedb "] name = "NeuroTreeModels" uuid = "1db4e0a5-a364-4b0c-897c-2bd5a4a3a1f2" -version = "1.1.1" +version = "1.1.2" [deps] CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" @@ -24,13 +24,13 @@ DataFrames = "1.3" Flux = "0.13, 0.14" MLJModelInterface = "1.0" MLUtils = "0.4" -Optimisers = "0.2, 0.3" +Optimisers = "0.3" Random = "1" Statistics = "1" StatsBase = "0.34" Tables = "1.9" cuDNN = "1" -julia = "1.6" +julia = "1.9" [extras] CategoricalArrays = "324d7699-5711-5eae-9e2f-1d82baa6b597" diff --git a/README.md b/README.md index e7e66f8..8a6234e 100644 --- a/README.md +++ b/README.md @@ -17,10 +17,12 @@ ## Installation -```julia +```julia-repl ] add NeuroTreeModels ``` +⚠ Compatible with Julia >= v1.9. + ## Configuring a model A model configuration is defined with the [NeuroTreeRegressor](https://evovest.github.io/NeuroTreeModels.jl/dev/models#NeuroTreeModels.NeuroTreeRegressor) constructor: diff --git a/src/MLJ.jl b/src/MLJ.jl index 95c1c4c..78da31a 100644 --- a/src/MLJ.jl +++ b/src/MLJ.jl @@ -68,9 +68,6 @@ function MMI.predict(::NeuroTreeRegressor, fitresult, A) return pred end -MMI.supports_weights(::Type{<:NeuroTreeRegressor}) = true - - # Metadata MMI.metadata_pkg.( (NeuroTreeRegressor),