CI errors be gone #1767

Job	Run time
GPU Tests	4m 22s
python 3.9 - Lint	1m 0s
python 3.9 - Type check	3m 37s
python 3.9 - Build	1m 12s
python 3.9 - Style	21s
python 3.9 - Docs	2m 36s
python 3.9 - Test	2m 55s
python 3.9 - Datasets integration	34s
python 3.9 - PyTorch integration	38s
python 3.9 - Transformers integration	3m 59s
python 3.9 - FairScale integration	48s
python 3.9 - W&B integration	3m 18s
python 3.9 - Beaker integration	2m 56s
python 3.9 - Flax integration	4m 44s
python 3.9 - GS integration	44s
python 3.9 - Example - train_lm	1m 27s
python 3.8 - Test	3m 20s
python 3.10 - Test	3m 4s
Release	0s
	41m 35s

Provide feedback