The raw dataset is TFRecord format: https://github.com/google-research/google-research/blob/master/android_control/README.md

And in your test script, it seems you have done some process on the original data, can you share the scripts?