Add cache_dir to save features TextDataset
This is in case the dataset is in a RO filesystem, for which is the case
in tests (GKE TPU tests).

Thanks for the quick reviews @LysandreJik and @sgugger! And yep updating black version did it (I think) thanks.