[Enhance] Make scipy as a default dependency in runtime in dev-1.x

pull/2362/head
MengzhangLI 2022-11-30 21:51:38 +08:00
parent e98231ba5f
commit 925110044c
2 changed files with 10 additions and 4 deletions

View File

@ -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

View File

@ -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'])