add iteration_parameter

oops

Author: ablaom

src/StatisticalTraits.jl

@@ -31,7 +31,8 @@ const TRAITS = [
     :reports_each_observation,
     :aggregation,
     :is_feature_dependent,
-    :distribution_type
+    :distribution_type,
+    :iteration_parameter
 ]
 
 
@@ -150,7 +151,7 @@ supports_class_weights(::Type) = false
 # Possible values of the following are `:deterministic`, `:probabilistic`,
 # `:interval`, or `:unknown`:
 
-prediction_type(::Type)        = :unknown 
+prediction_type(::Type)        = :unknown
 
 # Miscellaneous:
 
@@ -164,6 +165,7 @@ aggregation(::Type)     = Mean()
 is_feature_dependent(::Type)     = false
 reports_each_observation(::Type) = false
 distribution_type(::Type)        = missing
+iteration_parameter(::Type)      = nothing
 
 # Returns a tuple, with one entry per field of `T` (the type of some
 # statistical model, for example). Each entry is `nothing` or defines

test/runtests.jl

@@ -83,7 +83,8 @@ const NONCONSTANT = [:docstring,
                      :name,
                      :human_name,
                      :hyperparameters,
-                     :hyperparameter_types]
+                     :hyperparameter_types,
+                     :iteration_parameter]
 
 @testset "traits with constant fall-back" begin
     for trait in setdiff(StatisticalTraits.TRAITS, NONCONSTANT)