Tensorflow 2.0: flat_map() to flatten Dataset of Dataset returns cardinality -2

Question

I am trying to run the following code (as given in Tensorflow documentation) to create windows of my data and then flatten the dataset of datasets.

window_size =