fix: mark update_class_weights_in_features transformation for version…

… 0.6 (#2481)
ludwig-ai · Sep 14, 2022 · 230383a · 230383a
1 parent 9aa72c9
commit 230383a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/utils/backward_compatibility.py b/ludwig/utils/backward_compatibility.py
@@ -161,7 +161,7 @@ def _traverse_dicts(config: Any, f: Callable[[Dict], None]):
             _traverse_dicts(v, f)
 
 
-@register_config_transformation("0.4", ["output_features"])
+@register_config_transformation("0.6", ["output_features"])
 def update_class_weights_in_features(feature: Dict[str, Any]) -> Dict[str, Any]:
     if LOSS in feature:
         class_weights = feature[LOSS].get(CLASS_WEIGHTS, None)