🐛 Random Resampling should allow categorical types as well

JuliaAI · Jan 9, 2024 · b4c7ef9 · b4c7ef9
1 parent 031af23
commit b4c7ef9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/oversampling_methods/random_oversample/interface_mlj.jl b/src/oversampling_methods/random_oversample/interface_mlj.jl
@@ -41,8 +41,8 @@ MMI.metadata_model(
   RandomOversampler,
     input_scitype = Tuple{
                         Union{
-                            Table(Continuous),
-                            AbstractMatrix{Continuous}
+                            Table(Union{Infinite, Finite}),
+                            AbstractMatrix{Infinite}
                         }, 
                         AbstractVector
                     },

diff --git a/src/undersampling_methods/random_undersample/interface_mlj.jl b/src/undersampling_methods/random_undersample/interface_mlj.jl
@@ -52,8 +52,8 @@ MMI.metadata_model(
     RandomUndersampler,
     input_scitype = Tuple{
                         Union{
-                            Table(Continuous),
-                            AbstractMatrix{Continuous}
+                            Table(Union{Infinite, Finite}),
+                            AbstractMatrix{Infinite}
                         }, 
                         AbstractVector
                     },