Xander Schauffele is set for a return to the PGA Tour at the Arnold Palmer Invitational at Bay Hill after almost two months out with a rib injury.