add supports_class_weights to MODEL_TRAITS

Author: ablaom

src/MLJModelInterface.jl

@@ -1,13 +1,27 @@
 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]
+    :input_scitype,
+    :output_scitype,
+    :target_scitype,
+    :is_pure_julia,
+    :package_name,
+    :package_license,
+    :load_path,
+    :package_uuid,
+    :package_url,
+    :is_wrapper,
+    :supports_weights,
+    :supports_class_weights,
+    :supports_online,
+    :docstring,
+    :name,
+    :is_supervised,
+    :prediction_type,
+    :implemented_methods,
+    :hyperparameters,
+    :hyperparameter_types,
+    :hyperparameter_ranges]
 
 # ------------------------------------------------------------------------
 # Dependencies (ScientificTypes and StatisticalTraits have none)

src/model_traits.jl

@@ -1,6 +1,4 @@
-## MODEL TRAITS
-
-# model trait names:
+## OVERLOADING TRAIT DEFAULTS RELEVANT TO MODELS
 
 StatisticalTraits.docstring(M::Type{<:MLJType}) = name(M)
 StatisticalTraits.docstring(M::Type{<:Model}) =