Fixup docker build err #5348

Job	Run time
run-tests (latest, test_prod)	3m 6s
run-tests (latest, test_core)	4m 23s
run-tests (latest, test_cli)	4m 14s
run-tests (latest, test_big_modeling)	4m 0s
run-tests (latest, test_deepspeed)	3m 49s
run-tests (latest, test_fsdp)	3m 53s
run-tests (latest, test_example_differences)	4m 0s
run-tests (latest, test_checkpoint_step)	5m 24s
run-tests (latest, test_checkpoint_epoch)	5m 11s
run-tests (latest, test_rest)	12m 44s
run-tests (minimum, test_prod)	7m 9s
run-tests (minimum, test_core)	7m 10s
run-tests (minimum, test_cli)	6m 30s
run-tests (minimum, test_big_modeling)	6m 23s
run-tests (minimum, test_deepspeed)	8m 36s
run-tests (minimum, test_fsdp)	6m 19s
run-tests (minimum, test_example_differences)	6m 8s
run-tests (minimum, test_checkpoint_step)	9m 18s
run-tests (minimum, test_checkpoint_epoch)	7m 38s
run-tests (minimum, test_rest)	15m 5s
	2h 11m 0s

Provide feedback