FIX backward compatibility was broken
MISC: deprecate balance_weights (it's internal)