Yide Shentu

https://github.com/google-research/language-table/blob/main/language_table/train/input_pipeline_rlds.py#L256 See code here I don't think the `tf.nest.flatten` will preserve the keyvalue order and in fact, if you run the call this dataset creation method `input_pipeline_rlds.create_datasets` the dataloader it...