fix import mmrazor error in mmcls test (open-mmlab#467)

humu789 · May 13, 2022 · 6eb83a9 · 6eb83a9
1 parent 9a4028b
commit 6eb83a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_codebase/test_mmcls/test_classification.py b/tests/test_codebase/test_mmcls/test_classification.py
@@ -64,7 +64,8 @@ def test_init_pytorch_model(from_mmrazor: Any):
             _ = _task_processor.from_mmrazor
         return
     assert from_mmrazor == _task_processor.from_mmrazor
-
+    if from_mmrazor:
+        pytest.importorskip('mmrazor', reason='mmrazor is not installed.')
     model = _task_processor.init_pytorch_model(None)
     assert isinstance(model, BaseClassifier)