From 08cc36ddff49f97b5c9e34af9d96d24aef847ef2 Mon Sep 17 00:00:00 2001
From: Anton Lozhkov <anton@huggingface.co>
Date: Sat, 17 Dec 2022 22:57:30 +0100
Subject: [PATCH] Fix MPS fast test warnings (#1744)

* unset level
---
 tests/test_pipelines_common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_pipelines_common.py b/tests/test_pipelines_common.py
index 5d5c69c5c7..e51e6737d4 100644
--- a/tests/test_pipelines_common.py
+++ b/tests/test_pipelines_common.py
@@ -233,11 +233,11 @@ class PipelineTesterMixin:
 
         output = pipe(**inputs)
 
+        logger.setLevel(level=diffusers.logging.WARNING)
         if torch_device != "mps":
             # TODO(Pedro) - not sure why, but not at all reproducible at the moment it seems
             # make sure that batched and non-batched is identical
             assert np.abs(output_batch[0][0] - output[0][0]).max() < 1e-4
-            logger.setLevel(level=diffusers.logging.WARNING)
 
     def test_dict_tuple_outputs_equivalent(self):
         if torch_device == "mps" and self.pipeline_class in (