diff --git a/mmocr/core/__init__.py b/mmocr/core/__init__.py index 5c2da640..14a275f7 100644 --- a/mmocr/core/__init__.py +++ b/mmocr/core/__init__.py @@ -1,6 +1,5 @@ # Copyright (c) OpenMMLab. All rights reserved. from .data_structures import * # NOQA -from .loops import * # NOQA from .visualization import TextDetLocalVisualizer, TextRecogLocalVisualizer from .visualize import (det_recog_show_result, imshow_edge, imshow_node, imshow_pred_boundary, imshow_text_char_boundary, diff --git a/mmocr/engine/__init__.py b/mmocr/engine/__init__.py new file mode 100644 index 00000000..03ee0e85 --- /dev/null +++ b/mmocr/engine/__init__.py @@ -0,0 +1,2 @@ +# Copyright (c) OpenMMLab. All rights reserved. +from .runner import * # NOQA diff --git a/mmocr/core/loops/__init__.py b/mmocr/engine/runner/__init__.py similarity index 100% rename from mmocr/core/loops/__init__.py rename to mmocr/engine/runner/__init__.py diff --git a/mmocr/core/loops/multi_loops.py b/mmocr/engine/runner/multi_loops.py similarity index 100% rename from mmocr/core/loops/multi_loops.py rename to mmocr/engine/runner/multi_loops.py diff --git a/mmocr/utils/setup_env.py b/mmocr/utils/setup_env.py index 61f85bf1..b012f26a 100644 --- a/mmocr/utils/setup_env.py +++ b/mmocr/utils/setup_env.py @@ -18,6 +18,7 @@ def register_all_modules(init_default_scope: bool = True) -> None: """ # noqa import mmocr.core # noqa: F401,F403 import mmocr.datasets # noqa: F401,F403 + import mmocr.engine # noqa: F401,F403 import mmocr.evaluation # noqa: F401,F403 import mmocr.models # noqa: F401,F403 diff --git a/tests/test_core/test_loops/test_multi_loop.py b/tests/test_engine/test_runner/test_multi_loop.py similarity index 99% rename from tests/test_core/test_loops/test_multi_loop.py rename to tests/test_engine/test_runner/test_multi_loop.py index 46c33813..0efe89ba 100644 --- a/tests/test_core/test_loops/test_multi_loop.py +++ b/tests/test_engine/test_runner/test_multi_loop.py @@ -14,7 +14,7 @@ from mmengine.registry import DATASETS, HOOKS, METRICS, MODELS from mmengine.runner import Runner from torch.utils.data import Dataset -from mmocr.core.loops import MultiTestLoop, MultiValLoop +from mmocr.engine.runner import MultiTestLoop, MultiValLoop @MODELS.register_module()