Merge pull request #195 from JuliaAI/class-weights-fit-data-scitype

Tweak fit_data_scitype fallback to resolve issue with class weights
JuliaAI · Mar 22, 2024 · 5ac60f3 · 5ac60f3
2 parents 3e36cc7 + 12f588e
commit 5ac60f3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MLJModelInterface"
 uuid = "e80e1ace-859a-464e-9ed9-23947d8ae3ea"
 authors = ["Thibaut Lienart and Anthony Blaom"]
-version = "1.9.5"
+version = "1.9.6"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

diff --git a/src/model_traits.jl b/src/model_traits.jl
@@ -59,7 +59,7 @@ function supervised_fit_data_scitype(M)
         W = AbstractVector{<:Union{Continuous, Count}} # weight scitype
         return Union{ret, Tuple{I, T, W}}
     elseif supports_class_weights(M)
-        W = AbstractDict{Finite, <:Union{Continuous, Count}}
+        W = Any
         return Union{ret, Tuple{I, T, W}}
     end
     return ret