Allow sampler to take in images #4288

Job	Run time
Build tunix package / Build Tunix wheel	13s
tunix_cpu_unit_tests / run	1m 8s
tunix_tpu_unit_tests / run_dev	1m 8s
tunix_tpu_unit_tests / run_prod	1s
Notify failed build	3s
	2m 33s

Provide feedback