AutoMM: Fix crash due to undefined DataContainer (open-mmlab#2630)

* Fix crash due to undefined DataContainer * fixed lint error
FANGAreNotGnu · Jan 4, 2023 · b418cab · b418cab
1 parent d8272e0
commit b418cab
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/multimodal/src/autogluon/multimodal/utils/environment.py b/multimodal/src/autogluon/multimodal/utils/environment.py
@@ -8,12 +8,8 @@
 import torch
 from torch import nn
 
-try:
-    from mmcv.parallel import DataContainer
-except:
-    pass
-
 from ..constants import AUTOMM, OBJECT_DETECTION, OCR
+from .mmcv import DataContainer
 
 logger = logging.getLogger(AUTOMM)