Oh ok. I missed that. I was confused because in remove_label() you call upd_doc() then fit_label_estimator([doc], removed_label=).
It's ok for me then :)