https://github.com/EpistasisLab/tpot2/blob/8ebc82cb3b4d6fadfa5228a7de3134a9edc59acb/tpot2/tpot_estimator/estimator.py#L661 I think the wrong argument is being passed to the classification variable?