Key question is whether he was pressured to leave early than his planned March departure. Wray threatened to resign if McCabe was forced out.
I'm guessing things became so awkward with Trump and Sessions that he decided 'feck it, I don't need this' and opted to not stick around the extra month.