sklift.viz.plot_treatment_balance_curve