Fix issue with GDA data input pipeline where array shard indices would be placed in a set, thereby rearranging the order of the GDA shards (GDA requires a specific ordering without deviation.
Fix issue with GDA data input pipeline where array shard indices would be placed in a set, thereby rearranging the order of the GDA shards (GDA requires a specific ordering without deviation.

PiperOrigin-RevId: 474629073

google-research/t5x