Remove import search within current directory #1888

Job	Run time
python 3.9 - Lint	42s
python 3.9 - Type check	3m 26s
python 3.9 - Build	50s
python 3.9 - Style	17s
GPU Tests	12s
python 3.9 - Docs	1m 17s
python 3.9 - Test	2m 45s
python 3.9 - Datasets integration	42s
python 3.9 - PyTorch integration	32s
python 3.9 - Transformers integration	2m 44s
python 3.9 - FairScale integration	28s
python 3.9 - W&B integration	1m 59s
python 3.9 - Beaker integration	41s
python 3.9 - Flax integration	4m 20s
python 3.9 - GS integration	20s
python 3.9 - Example - train_lm	56s
python 3.8 - Test	3m 25s
python 3.10 - Test	3m 33s
Release	0s
	29m 9s

Provide feedback