From 925110044c89c52523313bb7448d0c1f1e5c1d9e Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Wed, 30 Nov 2022 21:51:38 +0800 Subject: [PATCH] [Enhance] Make scipy as a default dependency in runtime in dev-1.x --- tests/test_models/test_backbones/test_beit.py | 7 +++++-- tests/test_models/test_backbones/test_mae.py | 7 +++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/tests/test_models/test_backbones/test_beit.py b/tests/test_models/test_backbones/test_beit.py index cf3960894..59a12c5d0 100644 --- a/tests/test_models/test_backbones/test_beit.py +++ b/tests/test_models/test_backbones/test_beit.py @@ -140,8 +140,11 @@ def test_beit_init(): } } model = BEiT(img_size=(512, 512)) - with pytest.raises(AttributeError): - model.resize_rel_pos_embed(ckpt) + # If scipy is installed, this AttributeError would not be raised. + from mmengine.utils import is_installed + if not is_installed('scipy'): + with pytest.raises(AttributeError): + model.resize_rel_pos_embed(ckpt) # pretrained=None # init_cfg=123, whose type is unsupported diff --git a/tests/test_models/test_backbones/test_mae.py b/tests/test_models/test_backbones/test_mae.py index 562d067a7..16f52b54b 100644 --- a/tests/test_models/test_backbones/test_mae.py +++ b/tests/test_models/test_backbones/test_mae.py @@ -138,8 +138,11 @@ def test_mae_init(): } } model = MAE(img_size=(512, 512)) - with pytest.raises(AttributeError): - model.resize_rel_pos_embed(ckpt) + # If scipy is installed, this AttributeError would not be raised. + from mmengine.utils import is_installed + if not is_installed('scipy'): + with pytest.raises(AttributeError): + model.resize_rel_pos_embed(ckpt) # test resize abs pos embed ckpt = model.resize_abs_pos_embed(ckpt['state_dict'])