Fixed minor lintint errors. flake8 passes.

Author: anguelos

test/test_transforms.py

@@ -205,15 +205,13 @@ def test_random_crop(self):
         assert result.size(2) == width
         assert np.allclose(img.numpy(), result.numpy())
 
-        lheight = height + 1
-        lwidth = width + 1
         result = transforms.Compose([
             transforms.ToPILImage(),
-            transforms.RandomCrop((lheight, lwidth), pad_if_needed=True),
+            transforms.RandomCrop((height + 1, width + 1), pad_if_needed=True),
             transforms.ToTensor(),
         ])(img)
-        assert result.size(1) == lheight
-        assert result.size(2) == lwidth
+        assert result.size(1) == height + 1
+        assert result.size(2) == width + 1
 
     def test_pad(self):
         height = random.randint(10, 32) * 2
@@ -1174,4 +1172,4 @@ def test_random_grayscale(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

torchvision/transforms/transforms.py

@@ -416,7 +416,7 @@ def __call__(self, img):
             img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0))
         # pad the height if needed
         if self.pad_if_needed and img.size[1] < self.size[0]:
-            img = F.pad(img,(0, int((1 + self.size[0] - img.size[1]) / 2)))
+            img = F.pad(img, (0, int((1 + self.size[0] - img.size[1]) / 2)))
 
         i, j, h, w = self.get_params(img, self.size)