fixing dictionary generation for squad dataset.
fixing dictionary generation for squad dataset.

PiperOrigin-RevId: 226181683

tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets,tensorflow/datasets