diff --git a/tests/test_utils/test_lazy_init/utils.py b/tests/test_utils/test_lazy_init/lazy_init_utils.py similarity index 100% rename from tests/test_utils/test_lazy_init/utils.py rename to tests/test_utils/test_lazy_init/lazy_init_utils.py diff --git a/tests/test_utils/test_lazy_init/test_distribute.py b/tests/test_utils/test_lazy_init/test_distribute.py index c15b055e8..fd91e7e91 100644 --- a/tests/test_utils/test_lazy_init/test_distribute.py +++ b/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 diff --git a/tests/test_utils/test_lazy_init/test_models.py b/tests/test_utils/test_lazy_init/test_models.py index 4a0217b31..f828b23a9 100644 --- a/tests/test_utils/test_lazy_init/test_models.py +++ b/tests/test_utils/test_lazy_init/test_models.py @@ -1,5 +1,5 @@ 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