Gandhi had envisaged the British troops remaining in India after independence for some time to train Indians. That is, Gandhi never considered driving out the British as an option, in which case the British would certainly not have obliged by remaining in India to train their adversaries.
