fix style

Signed-off-by: Yao Matrix <[email protected]>

Author: yao-matrix

tests/pipelines/dance_diffusion/test_dance_diffusion.py

@@ -20,7 +20,14 @@
 import torch
 
 from diffusers import DanceDiffusionPipeline, IPNDMScheduler, UNet1DModel
-from diffusers.utils.testing_utils import backend_empty_cache, enable_full_determinism, nightly, require_torch_accelerator, require_torch_gpu, skip_mps, torch_device
+from diffusers.utils.testing_utils import (
+    backend_empty_cache,
+    enable_full_determinism,
+    nightly,
+    require_torch_accelerator,
+    skip_mps,
+    torch_device,
+)
 
 from ..pipeline_params import UNCONDITIONAL_AUDIO_GENERATION_BATCH_PARAMS, UNCONDITIONAL_AUDIO_GENERATION_PARAMS
 from ..test_pipelines_common import PipelineTesterMixin

tests/pipelines/latent_diffusion/test_latent_diffusion.py

@@ -27,7 +27,6 @@
     load_numpy,
     nightly,
     require_torch_accelerator,
-    require_torch_gpu,
     torch_device,
 )
 

tests/pipelines/musicldm/test_musicldm.py

@@ -39,7 +39,13 @@
     UNet2DConditionModel,
 )
 from diffusers.utils import is_xformers_available
-from diffusers.utils.testing_utils import backend_empty_cache, enable_full_determinism, nightly, require_torch_accelerator, require_torch_gpu, torch_device
+from diffusers.utils.testing_utils import (
+    backend_empty_cache,
+    enable_full_determinism,
+    nightly,
+    require_torch_accelerator,
+    torch_device,
+)
 
 from ..pipeline_params import TEXT_TO_AUDIO_BATCH_PARAMS, TEXT_TO_AUDIO_PARAMS
 from ..test_pipelines_common import PipelineTesterMixin

tests/pipelines/shap_e/test_shap_e.py

@@ -21,7 +21,13 @@
 
 from diffusers import HeunDiscreteScheduler, PriorTransformer, ShapEPipeline
 from diffusers.pipelines.shap_e import ShapERenderer
-from diffusers.utils.testing_utils import backend_empty_cache, load_numpy, nightly, require_torch_accelerator, require_torch_gpu, torch_device
+from diffusers.utils.testing_utils import (
+    backend_empty_cache,
+    load_numpy,
+    nightly,
+    require_torch_accelerator,
+    torch_device,
+)
 
 from ..test_pipelines_common import PipelineTesterMixin, assert_mean_pixel_difference
 

tests/pipelines/shap_e/test_shap_e_img2img.py

@@ -29,7 +29,6 @@
     load_numpy,
     nightly,
     require_torch_accelerator,
-    require_torch_gpu,
     torch_device,
 )