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

Tweak fit_data_scitype fallback to resolve issue with class weights

Author: ablaom

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"

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