Add remove size one func to eval dataloader#3754

copybara-service[bot] merged 1 commit intomainAI-Hypercomputer/maxtext:mainfrom

chengnuojin-eval-shardingAI-Hypercomputer/maxtext:chengnuojin-eval-shardingCopy head branch name to clipboard