Re-enable skipped test

Signed-off-by: lizz <lizz@sensetime.com>
pull/2/head
lizz 2021-04-05 20:39:48 +08:00
parent 2ad699e099
commit b153cc01c8
3 changed files with 0 additions and 4 deletions

View File

@ -261,7 +261,6 @@ def test_psenet(cfg_file):
detector.show_result(img, results)
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
@pytest.mark.parametrize('cfg_file', [
'textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py',
'textdet/dbnet/dbnet_r50dcnv2_fpnc_1200e_icdar2015.py'

View File

@ -18,7 +18,6 @@ def test_char_attn():
assert out_feat_map.shape == torch.Size([1, 128, 32, 32])
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
def test_feat_generator():
in_feat = torch.rand(1, 128, 32, 32)
feat_generator = FeatGenerator(in_channels=128, out_channels=128)
@ -28,7 +27,6 @@ def test_feat_generator():
assert feat_map.shape == torch.Size([1, 128, 32, 32])
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
def test_cafcn_neck():
in_s1 = torch.rand(1, 64, 64, 64)
in_s2 = torch.rand(1, 128, 32, 32)

View File

@ -81,7 +81,6 @@ def test_base_recognizer():
tmp_dir.cleanup()
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
def test_seg_recognizer():
tmp_dir = tempfile.TemporaryDirectory()
# create dummy data