fix UT

open-mmlab · Jun 14, 2023 · 7d32b9a · 7d32b9a
1 parent f749743
commit 7d32b9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet/models/layers/transformer/hybrid_encoder.py b/mmdet/models/layers/transformer/hybrid_encoder.py
@@ -16,7 +16,7 @@
 try:
     from mmpretrain.models.backbones.repvgg import RepVGGBlock as _RepVGGBlock
 except ImportError:
-    RepVGGBlock = None
+    _RepVGGBlock = None
 
 
 class RepVGGBlock(_RepVGGBlock):
@@ -98,7 +98,7 @@ def __init__(self,
                  norm_cfg: OptConfigType = dict(type='BN', requires_grad=True),
                  act_cfg: OptConfigType = dict(type='SiLU', inplace=True)):
         super(CSPRepLayer, self).__init__()
-        assert RepVGGBlock is not None, (
+        assert _RepVGGBlock is not None, (
             'Please install mmpretrain to use CSPRepLayer')
         hidden_channels = int(out_channels * expansion)
         self.conv1 = ConvModule(