Police unit to train for attack at high school