add annotation to data loader next_batch #6690

Job	Run time
build_test (linux.2xlarge, 3.10, py310) / linux-job	6m 45s
build_test (linux.2xlarge, 3.8, py38) / linux-job	6m 50s
build_test (linux.2xlarge, 3.11, py311) / linux-job	7m 9s
build_test (linux.2xlarge, 3.9, py39) / linux-job	6m 57s
	27m 41s

Provide feedback