Skip to content

Commit 544d7fc

Browse files
committed
make sure fit_data_scitype is a Tuple
1 parent d29e1a7 commit 544d7fc

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/model_traits.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ for M in ABSTRACT_MODEL_SUBTYPES
2121
@eval(StatisticalTraits.abstract_type(::Type{<:$M}) = $M)
2222
end
2323

24-
StatisticalTraits.fit_data_scitype(M::Type{<:Model}) = input_scitype(M)
24+
StatisticalTraits.fit_data_scitype(M::Type{<:Model}) = Tuple{input_scitype(M)}
2525
StatisticalTraits.fit_data_scitype(::Type{<:Static}) = Tuple{}
2626
function StatisticalTraits.fit_data_scitype(M::Type{<:Supervised})
2727
I = input_scitype(M)

test/model_traits.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ end
138138

139139
@test fit_data_scitype(P2()) ==
140140
Tuple{Table(Continuous),AbstractVector{<:Multiclass}}
141-
@test fit_data_scitype(U2()) == Table(Continuous)
141+
@test fit_data_scitype(U2()) == Tuple{Table(Continuous)}
142142
@test fit_data_scitype(S2()) == Tuple{}
143143
end
144144

0 commit comments

Comments
 (0)