From c77954bb90e60d8ff4b07fa1b1559d57bb9b846e Mon Sep 17 00:00:00 2001
From: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
Date: Mon, 10 Apr 2023 14:56:52 +0800
Subject: [PATCH] [Fix] Fix import error (#746)

* fix import error

* update
---
 mmselfsup/registry.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/mmselfsup/registry.py b/mmselfsup/registry.py
index 5b2b50b7..c567872d 100644
--- a/mmselfsup/registry.py
+++ b/mmselfsup/registry.py
@@ -55,15 +55,14 @@ __all__ = [
 # Registries For Runner and the related
 # manage all kinds of runners like `EpochBasedRunner` and `IterBasedRunner`
 RUNNERS = Registry(
-    'runner', parent=MMENGINE_RUNNERS, locations=['mmselfsup.engine.runner'])
+    'runner', parent=MMENGINE_RUNNERS, locations=['mmselfsup.engine'])
 # manage runner constructors that define how to initialize runners
 RUNNER_CONSTRUCTORS = Registry(
     'runner constructor',
     parent=MMENGINE_RUNNER_CONSTRUCTORS,
-    locations=['mmselfsup.engine.runner'])
+    locations=['mmselfsup.engine'])
 # manage all kinds of loops like `EpochBasedTrainLoop`
-LOOPS = Registry(
-    'loop', parent=MMENGINE_LOOPS, locations=['mmselfsup.engine.runner'])
+LOOPS = Registry('loop', parent=MMENGINE_LOOPS, locations=['mmselfsup.engine'])
 # manage all kinds of hooks like `CheckpointHook`
 HOOKS = Registry(
     'hook', parent=MMENGINE_HOOKS, locations=['mmselfsup.engine.hooks'])