remove problematic unit test

Author: phisanti

tests/test_pixelunshuffle.py

@@ -46,5 +46,6 @@ def test_inverse_operation(self):
         unshuffled = pixelunshuffle(shuffled, spatial_dims=3, scale_factor=2)
         torch.testing.assert_close(x, unshuffled)
 
+
 if __name__ == "__main__":
     unittest.main()

tests/test_restormer.py

@@ -123,7 +123,7 @@ class TestRestormer(unittest.TestCase):
     @skipUnless(has_einops, "Requires einops")
     @parameterized.expand(TEST_CASES_RESTORMER)
     def test_shape(self, input_param, input_shape, expected_shape):
-        if input_param.get('flash_attention', False) and not torch.cuda.is_available():
+        if input_param.get("flash_attention", False) and not torch.cuda.is_available():
             self.skipTest("Flash attention requires CUDA")
         net = Restormer(**input_param)
         with eval_mode(net):