[Fix] Fix import problem in __init__ of the gcn directory (kennymck…

…ormick#114)
jeffreyking23 · Dec 23, 2022 · ba1d5ac · ba1d5ac
1 parent f1c68ec
commit ba1d5ac
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/pyskl/models/gcns/__init__.py b/pyskl/models/gcns/__init__.py
@@ -1,12 +1,7 @@
-from .aagcn import AAGCN
-from .ctrgcn import CTRGCN
-from .dgstgcn import DGSTGCN
-from .msg3d import MSG3D
-from .sgn import SGN
-from .stgcn import STGCN
-from .utils import mstcn, unit_aagcn, unit_gcn, unit_group_gcn, unit_tcn
-
-__all__ = [
-    'unit_gcn', 'unit_aagcn', 'unit_group_gcn', 'unit_tcn', 'mstcn',
-    'STGCN', 'AAGCN', 'MSG3D', 'CTRGCN', 'SGN', 'DGSTGCN'
-]
+from .aagcn import AAGCN  # noqa: F401, F403
+from .ctrgcn import CTRGCN  # noqa: F401, F403
+from .dgstgcn import DGSTGCN  # noqa: F401, F403
+from .msg3d import MSG3D  # noqa: F401, F403
+from .sgn import SGN  # noqa: F401, F403
+from .stgcn import STGCN  # noqa: F401, F403
+from .utils import *  # noqa: F401, F403