fix deepscaler notebook #4153

Job	Run time
Build tunix package / Build Tunix wheel	9s
tunix_cpu_unit_tests / run	2m 40s
tunix_tpu_unit_tests / run_dev	22m 45s
tunix_tpu_unit_tests / run_prod	12m 28s
Notify failed build	5s
	38m 7s

Provide feedback