From 827f5d66e578718e7780c1f8c8efa1a2640e052a Mon Sep 17 00:00:00 2001 From: Jirka Date: Sat, 6 May 2023 00:38:18 +0200 Subject: [PATCH] skps --- tests/image/semantic_segm/test_backbones.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/image/semantic_segm/test_backbones.py b/tests/image/semantic_segm/test_backbones.py index 9a76b60061..ba05e83c7a 100644 --- a/tests/image/semantic_segm/test_backbones.py +++ b/tests/image/semantic_segm/test_backbones.py @@ -17,10 +17,7 @@ from flash.image.segmentation.backbones import SEMANTIC_SEGMENTATION_BACKBONES -@pytest.mark.parametrize( - ["backbone"], - ["resnet50", "dpn131"], -) +@pytest.mark.parametrize("backbone", ["resnet50", "dpn131"]) @pytest.mark.skipif(not _SEGMENTATION_MODELS_AVAILABLE, reason="No SMP") def test_semantic_segmentation_backbones_registry(backbone): backbone = SEMANTIC_SEGMENTATION_BACKBONES.get(backbone)()