synched with StatisticalTraits gamma commit

forgotten

Author: ablaom

src/MLJModelInterface.jl

@@ -1,5 +1,14 @@
 module MLJModelInterface
 
+const MODEL_TRAITS = [
+    :input_scitype, :output_scitype, :target_scitype,
+    :is_pure_julia, :package_name, :package_license,
+    :load_path, :package_uuid, :package_url,
+    :is_wrapper, :supports_weights, :supports_online,
+    :docstring, :name, :is_supervised,
+    :prediction_type, :implemented_methods, :hyperparameters,
+    :hyperparameter_types, :hyperparameter_ranges]
+
 # ------------------------------------------------------------------------
 # Dependencies (ScientificTypes and StatisticalTraits have none)
 using ScientificTypes
@@ -29,13 +38,9 @@ export fit, update, update_data, transform, inverse_transform,
     predict_joint, evaluate, clean!, reformat
 
 # model traits
-export input_scitype, output_scitype, target_scitype,
-       is_pure_julia, package_name, package_license,
-       load_path, package_uuid, package_url,
-       is_wrapper, supports_weights, supports_online,
-       docstring, name, is_supervised,
-       prediction_type, implemented_methods, hyperparameters,
-       hyperparameter_types, hyperparameter_ranges
+for trait in MODEL_TRAITS
+    @eval(export $trait)
+end
 
 # data operations
 export matrix, int, classes, decoder, table,

src/model_traits.jl

@@ -1,24 +1,16 @@
 ## MODEL TRAITS
 
 # model trait names:
-const MODEL_TRAITS = [
-    :input_scitype, :output_scitype, :target_scitype,
-    :is_pure_julia, :package_name, :package_license,
-    :load_path, :package_uuid, :package_url,
-    :is_wrapper, :supports_weights, :supports_online,
-    :docstring, :name, :is_supervised,
-    :prediction_type, :implemented_methods, :hyperparameters,
-    :hyperparameter_types, :hyperparameter_ranges]
 
-StatTraits.docstring(M::Type{<:MLJType}) = name(M)
-StatTraits.docstring(M::Type{<:Model}) =
+StatisticalTraits.docstring(M::Type{<:MLJType}) = name(M)
+StatisticalTraits.docstring(M::Type{<:Model}) =
     "$(name(M)) from $(package_name(M)).jl.\n" *
     "[Documentation]($(package_url(M)))."
 
-StatTraits.is_supervised(::Type{<:Supervised})      = true
-StatTraits.prediction_type(::Type{<:Deterministic}) = :deterministic
-StatTraits.prediction_type(::Type{<:Probabilistic}) = :probabilistic
-StatTraits.prediction_type(::Type{<:Interval})      = :interval
+StatisticalTraits.is_supervised(::Type{<:Supervised})      = true
+StatisticalTraits.prediction_type(::Type{<:Deterministic}) = :deterministic
+StatisticalTraits.prediction_type(::Type{<:Probabilistic}) = :probabilistic
+StatisticalTraits.prediction_type(::Type{<:Interval})      = :interval
 
 # implementation is deferred as it requires methodswith which depends upon
 # InteractiveUtils which we don't want to bring here as a dependency