skps

Lightning-Universe · May 5, 2023 · 827f5d6 · 827f5d6
1 parent 26ea844
commit 827f5d6
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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)()