isolate inference needed for training the dependency parser