also added trait

Author: pat-alt

src/model_traits.jl

@@ -27,6 +27,7 @@ StatTraits.is_supervised(::Type{<:SupervisedAnnotator}) = true
 StatTraits.prediction_type(::Type{<:Deterministic}) = :deterministic
 StatTraits.prediction_type(::Type{<:Probabilistic}) = :probabilistic
 StatTraits.prediction_type(::Type{<:Interval}) = :interval
+StatTraits.prediction_type(::Type{<:ProbabilisticSet}) = :probabistic_set
 StatTraits.prediction_type(::Type{<:ProbabilisticDetector}) = :probabilistic
 StatTraits.prediction_type(::Type{<:DeterministicDetector}) = :deterministic