CI errors be gone (#601) #1778

Job	Run time
GPU Tests	3m 37s
python 3.9 - Lint	3m 15s
python 3.9 - Type check	6m 17s
python 3.9 - Build	3m 42s
python 3.9 - Style	57s
python 3.9 - Docs	4m 44s
python 3.9 - Test	3m 41s
python 3.9 - Datasets integration	1m 19s
python 3.9 - PyTorch integration	1m 37s
python 3.9 - Transformers integration	4m 22s
python 3.9 - FairScale integration	1m 45s
python 3.9 - W&B integration	5m 45s
python 3.9 - Beaker integration	3m 33s
python 3.9 - Flax integration	6m 44s
python 3.9 - GS integration	1m 7s
python 3.9 - Example - train_lm	3m 16s
python 3.8 - Test	3m 57s
python 3.10 - Test	3m 44s
Release	0s
	1h 3m 22s

Provide feedback