correct xfail

sayakpaul · sayakpaul · commit 82571a42cfe1 · 2024-11-20T19:25:41.000+05:30
diff --git a/tests/lora/test_lora_layers_cogvideox.py b/tests/lora/test_lora_layers_cogvideox.py
@@ -127,7 +127,7 @@ def get_dummy_inputs(self, with_generator=True):
         return noise, input_ids, pipeline_inputs
 
     @skip_mps
-    @pytest.mark.xfail("Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)
+    @pytest.mark.xfail(reason="Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)
     def test_lora_fuse_nan(self):
         for scheduler_cls in self.scheduler_classes:
             components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)
diff --git a/tests/lora/utils.py b/tests/lora/utils.py
@@ -1511,7 +1511,7 @@ def test_simple_inference_with_text_denoiser_multi_adapter_weighted(self):
             )
 
     @skip_mps
-    @pytest.mark.xfail("Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)
+    @pytest.mark.xfail(reason="Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)
     def test_lora_fuse_nan(self):
         for scheduler_cls in self.scheduler_classes:
             components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)

Original file line number	Diff line number	Diff line change
`@@ -1511,7 +1511,7 @@ def test_simple_inference_with_text_denoiser_multi_adapter_weighted(self):`
`1511`	`1511`	`)`
`1512`	`1512`
`1513`	`1513`	`@skip_mps`
`1514`		`- @pytest.mark.xfail("Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)`
	`1514`	`+ @pytest.mark.xfail(reason="Test currently fails on CPU and PyTorch 2.5.1 but not on PyTorch 2.4.1.", strict=True)`
`1515`	`1515`	`def test_lora_fuse_nan(self):`
`1516`	`1516`	`for scheduler_cls in self.scheduler_classes:`
`1517`	`1517`	`components, text_lora_config, denoiser_lora_config = self.get_dummy_components(scheduler_cls)`