github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/3815/head
Frank Lee 2023-05-23 11:57:15 +08:00 committed by GitHub
parent ad93c736ea
commit 615e2e5fc1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
tests/test_utils/test_lazy_init/utils.py → tests/test_utils/test_lazy_init/lazy_init_utils.py
View File

2
tests/test_utils/test_lazy_init/test_distribute.py
View File

@ -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

2
tests/test_utils/test_lazy_init/test_models.py
View File

@ -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