[test] fixed lazy init test import error (#3799)

tests/test_utils/test_lazy_init/test_distribute.py

@@ -15,7 +15,7 @@ try:
     from colossalai.utils.model.experimental import LazyInitContext, LazyTensor, _MyTensor
 except:
     pass
-from utils import SUPPORT_LAZY, assert_dist_model_equal, set_seed
+from lazy_init_utils import SUPPORT_LAZY, assert_dist_model_equal, set_seed
 
 from tests.kit.model_zoo import model_zoo
 

tests/test_utils/test_lazy_init/test_models.py

 import pytest
-from utils import SUPPORT_LAZY, check_lazy_init
+from lazy_init_utils import SUPPORT_LAZY, check_lazy_init
 
 from tests.kit.model_zoo import model_zoo