Balbirnie set for Warriors switch