Allow sampler to take in images #4285

Job	Run time
Build tunix package / Build Tunix wheel	15s
tunix_cpu_unit_tests / run	1m 18s
tunix_tpu_unit_tests / run_prod	1m 55s
tunix_tpu_unit_tests / run_dev	10m 40s
Notify failed build	3s
	14m 11s

Provide feedback