Fix bug in to_tf_dataset
I'm going to merge it, but yeah - hopefully the CI runner just cleans that up automatically and few other people run the tests on Windows anyway!
Replace `set_format()` to `with_format()` so that we don't alter the original dataset in `to_tf_dataset()`