From 615e2e5fc1ec2211364e7a5f9e62156f69e7b984 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Tue, 23 May 2023 11:57:15 +0800
Subject: [PATCH] [test] fixed lazy init test import error (#3799)

---
 .../test_utils/test_lazy_init/{utils.py => lazy_init_utils.py}  | 0
 tests/test_utils/test_lazy_init/test_distribute.py              | 2 +-
 tests/test_utils/test_lazy_init/test_models.py                  | 2 +-
 3 files changed, 2 insertions(+), 2 deletions(-)
 rename tests/test_utils/test_lazy_init/{utils.py => lazy_init_utils.py} (100%)

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