diff --git a/tests/test_models/test_detector.py b/tests/test_models/test_detector.py index f83ae9cf..b0c13fa5 100644 --- a/tests/test_models/test_detector.py +++ b/tests/test_models/test_detector.py @@ -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' diff --git a/tests/test_models/test_ocr_neck.py b/tests/test_models/test_ocr_neck.py index 8af3e971..53f46d61 100644 --- a/tests/test_models/test_ocr_neck.py +++ b/tests/test_models/test_ocr_neck.py @@ -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) diff --git a/tests/test_models/test_recognizer.py b/tests/test_models/test_recognizer.py index 54f66ca6..29f4ded0 100644 --- a/tests/test_models/test_recognizer.py +++ b/tests/test_models/test_recognizer.py @@ -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