adapt style

Author: faaany

tests/models/autoencoders/test_models_vae.py

@@ -376,10 +376,12 @@ def prepare_init_args_and_inputs_for_common(self):
         return self.init_dict, self.inputs_dict()
 
     @unittest.skip
-    def test_training(self): ...
+    def test_training(self):
+        ...
 
     @unittest.skip
-    def test_ema_training(self): ...
+    def test_ema_training(self):
+        ...
 
 
 class AutoencoderKLTemporalDecoderFastTests(ModelTesterMixin, unittest.TestCase):

tests/pipelines/amused/test_amused.py

@@ -124,7 +124,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
         self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
 
     @unittest.skip("aMUSEd does not support lists of generators")
-    def test_inference_batch_single_identical(self): ...
+    def test_inference_batch_single_identical(self):
+        ...
 
 
 @slow

tests/pipelines/amused/test_amused_img2img.py

@@ -126,7 +126,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
         self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
 
     @unittest.skip("aMUSEd does not support lists of generators")
-    def test_inference_batch_single_identical(self): ...
+    def test_inference_batch_single_identical(self):
+        ...
 
 
 @slow

tests/pipelines/amused/test_amused_inpaint.py

@@ -130,7 +130,8 @@ def test_inference_batch_consistent(self, batch_sizes=[2]):
         self._test_inference_batch_consistent(batch_sizes=batch_sizes, batch_generator=False)
 
     @unittest.skip("aMUSEd does not support lists of generators")
-    def test_inference_batch_single_identical(self): ...
+    def test_inference_batch_single_identical(self):
+        ...
 
 
 @slow

tests/single_file/test_model_controlnet_single_file.py

@@ -49,7 +49,7 @@ def tearDown(self):
         super().tearDown()
         gc.collect()
         backend_empty_cache(torch_device)
-        
+
     def test_single_file_components(self):
         model = self.model_class.from_pretrained(self.repo_id)
         model_single_file = self.model_class.from_single_file(self.ckpt_path)

tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py

@@ -13,6 +13,7 @@
     numpy_cosine_similarity_distance,
     require_torch_accelerator,
     slow,
+    torch_device,
 )
 
 from .single_file_testing_utils import (