Build MNIST with one_hot=True in more cases