DOC: add balance_weights to reference