mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
MengzhangLI 2022-11-30 21:51:38 +08:00
parent e98231ba5f
commit 925110044c
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_models/test_backbones/test_beit.py
View File

@ -140,6 +140,9 @@ def test_beit_init():
}
}
model = BEiT(img_size=(512, 512))
# 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)

3
tests/test_models/test_backbones/test_mae.py
View File

@ -138,6 +138,9 @@ def test_mae_init():
}
}
model = MAE(img_size=(512, 512))
# 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)