Bug：name 'treatment' is not defined

[Simonleedata]

When I use the plot_uplift_preds fuction，it always reminds me : NameError: name 'treatment' is not defined. All the verision i have tried got the same result.

treatment_preds = tm.trmnt_preds_
control_preds = tm.ctrl_preds_
plot_uplift_preds(treatment_preds, control_preds);

---

NameError Traceback (most recent call last)
in
1 treatment_preds = tm.trmnt_preds_
2 control_preds = tm.ctrl_preds_
----> 3 plot_uplift_preds(treatment_preds, control_preds);

~/opt/anaconda3/lib/python3.8/site-packages/sklift/viz/base.py in plot_uplift_preds(trmnt_preds, ctrl_preds, log, bins)
29
30 # TODO: Add k as parameter: vertical line on plots
---> 31 check_consistent_length(trmnt_preds, ctrl_preds)
32
33 if not isinstance(bins, int) or bins <= 0:

NameError: name 'treatment' is not defined

[Simonleedata]

2 months ago, the code can got a correct result, but now, it can not work