You have name clash with the original `n_labels`, maybe `n_nonzero_labels` would dot the job.