No se esperaba que Yi entrara en el Draft de la NBA hasta el año 2009, dado que las autoridades chinas no autorizan a sus jugadores abandonar el país antes de los 22 años, pero a pesar de ello su equipo anunció que se presentaría al del 2007, donde fue elegido por Milwaukee Bucks en la sexta posición de la primera ronda.