FIX compute_class_weight edge case