diff --git a/tests/data/config/lazy_module_config/test_ast_transform.py b/tests/data/config/lazy_module_config/test_ast_transform.py index a8803dde24..6f0ada1736 100644 --- a/tests/data/config/lazy_module_config/test_ast_transform.py +++ b/tests/data/config/lazy_module_config/test_ast_transform.py @@ -3,7 +3,7 @@ from importlib.util import find_spec as find_module import numpy -import numpy.compat +import numpy.fft import numpy.linalg as linalg from mmengine.config import Config diff --git a/tests/test_config/test_lazy.py b/tests/test_config/test_lazy.py index d69822814b..1dda04fdaa 100644 --- a/tests/test_config/test_lazy.py +++ b/tests/test_config/test_lazy.py @@ -8,7 +8,7 @@ from unittest import TestCase import numpy -import numpy.compat +import numpy.fft import numpy.linalg as linalg from rich.progress import Progress @@ -56,17 +56,17 @@ def test_lazy_module(self): # 1.2 getattr as LazyAttr self.assertIsInstance(lazy_numpy.linalg, LazyAttr) - self.assertIsInstance(lazy_numpy.compat, LazyAttr) + self.assertIsInstance(lazy_numpy.fft, LazyAttr) - # 1.3 Build module from LazyObject. amp and functional can be accessed + # 1.3 Build module from LazyObject. linalg and fft can be accessed imported_numpy = lazy_numpy.build() self.assertIs(imported_numpy.linalg, linalg) - self.assertIs(imported_numpy.compat, numpy.compat) + self.assertIs(imported_numpy.fft, numpy.fft) # 1.4.1 Build module from LazyAttr imported_linalg = lazy_numpy.linalg.build() - imported_compat = lazy_numpy.compat.build() - self.assertIs(imported_compat, numpy.compat) + imported_fft = lazy_numpy.fft.build() + self.assertIs(imported_fft, numpy.fft) self.assertIs(imported_linalg, linalg) # 1.4.2 build class method from LazyAttr