STY: Fix linting

AnirudhDagar · AnirudhDagar · commit 95f9736a373b · 2021-08-26T07:26:48.000+05:30
diff --git a/test/test_tensor_creation_ops.py b/test/test_tensor_creation_ops.py
@@ -698,8 +698,8 @@ def test_cat_preserve_channels_last(self, device):
     def test_cat_out_memory_format(self, device):
         inp_size = (4, 4, 4, 4)
         expected_size = (8, 4, 4, 4)
-        a_cuda  = torch.randn(inp_size, device=device).contiguous(memory_format=torch.channels_last)
-        a_cpu  = torch.randn(inp_size, device='cpu').contiguous(memory_format=torch.channels_last)
+        a_cuda = torch.randn(inp_size, device=device).contiguous(memory_format=torch.channels_last)
+        a_cpu = torch.randn(inp_size, device='cpu').contiguous(memory_format=torch.channels_last)
         b_cuda = torch.randn(inp_size, device=device).contiguous(memory_format=torch.contiguous_format)
         b_cpu = torch.randn(inp_size, device='cpu').contiguous(memory_format=torch.contiguous_format)
         c_cuda = torch.randn(inp_size, device=device).contiguous(memory_format=torch.channels_last)
