Fix

Lightning-Universe · May 10, 2021 · 5f254b6 · 5f254b6
1 parent 2d76f37
commit 5f254b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/vision/segmentation/test_data.py b/tests/vision/segmentation/test_data.py
@@ -233,7 +233,7 @@ def test_from_files_warning(self, tmpdir):
         # instantiate the data module
 
         with pytest.raises(MisconfigurationException, match="The number of input files"):
-            dm = SemanticSegmentationData.from_files(
+            SemanticSegmentationData.from_files(
                 train_files=images,
                 train_targets=targets + [str(tmp_dir / "labels_img4.png")],
                 batch_size=2,