fix unit test

open-mmlab · Sep 15, 2022 · 8bd44a1 · 8bd44a1
1 parent 3c76adc
commit 8bd44a1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_model/test_model_utils.py b/tests/test_model/test_model_utils.py
@@ -41,7 +41,7 @@ def test_convert_syncbn():
     # Test convert to mmcv SyncBatchNorm
     if is_installed('mmcv'):
         # MMCV SyncBatchNorm is only supported on distributed training.
-        with pytest.raises(RuntimeError):
+        with pytest.raises((RuntimeError, AssertionError)):
             convert_sync_batchnorm(conv, implementation='mmcv')
 
     # Test convert to Pytorch SyncBatchNorm