fix import in transformer optimizer python script

fix import in transformer optimizer python script #22860