remove-redundent__init__

open-mmlab · Dec 29, 2022 · f0af246 · f0af246
1 parent 21acc17
commit f0af246
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/projects/PETR/petr/models/task_modules/__init__.py b/projects/PETR/petr/models/task_modules/__init__.py
@@ -1,11 +1,9 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-from mmdet.models.task_modules import AssignResult, BaseAssigner
-
 from .assigners import HungarianAssigner3D
 from .coders import NMSFreeCoder
 from .match_cost import BBox3DL1Cost, FocalLossCost, IoUCost
 
 __all__ = [
-    'BaseAssigner', 'AssignResult', 'HungarianAssigner3D', 'NMSFreeCoder',
-    'BBox3DL1Cost', 'IoUCost', 'FocalLossCost'
+    'HungarianAssigner3D', 'NMSFreeCoder', 'BBox3DL1Cost', 'IoUCost',
+    'FocalLossCost'
 ]
diff --git a/projects/PETR/petr/models/task_modules/assigners/__init__.py b/projects/PETR/petr/models/task_modules/assigners/__init__.py
@@ -1,9 +1,4 @@
 # Copyright (c) OpenMMLab. All rights reserved.
-from mmdet.models.task_modules import (AssignResult, BaseAssigner,
-                                       MaxIoUAssigner)
-
 from .hungarian_assigner_3d import HungarianAssigner3D
 
-__all__ = [
-    'BaseAssigner', 'MaxIoUAssigner', 'AssignResult', 'HungarianAssigner3D'
-]
+__all__ = ['HungarianAssigner3D']