[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Lightning-Universe · Jul 12, 2021 · d15d66b · d15d66b
1 parent 457989f
commit d15d66b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/image/test_backbones.py b/tests/image/test_backbones.py
@@ -33,10 +33,12 @@ def test_image_classifier_backbones_registry(backbone, expected_num_features):
 
 
 @pytest.mark.parametrize(["backbone", "pretrained", "expected_num_features"], [
-    pytest.param("resnet50", "supervised", 2048, marks=pytest.mark.skipif(
-        not _TORCHVISION_AVAILABLE, reason="No torchvision")),
-    pytest.param("resnet50", "simclr", 2048, marks=pytest.mark.skipif(
-        not _TORCHVISION_AVAILABLE, reason="No torchvision")),
+    pytest.param(
+        "resnet50", "supervised", 2048, marks=pytest.mark.skipif(not _TORCHVISION_AVAILABLE, reason="No torchvision")
+    ),
+    pytest.param(
+        "resnet50", "simclr", 2048, marks=pytest.mark.skipif(not _TORCHVISION_AVAILABLE, reason="No torchvision")
+    ),
 ])
 def test_pretrained_weights_registry(backbone, pretrained, expected_num_features):
     backbone_fn = IMAGE_CLASSIFIER_BACKBONES.get(backbone)