spell

Author: xadupre

_unittests/ut_torch_export_patches/test_patch_torch.py

@@ -363,7 +363,7 @@ def test_export_with_patch_tiny_llm_dim_meta(self):
 
         with self.subTest(input="no01", backed_size_oblivious=True):
             if not has_transformers("4.55"):
-                raise unittest.SkipIf("test not working with transformers<4.55")
+                raise unittest.SkipTest("test not working with transformers<4.55")
             with (
                 torch.fx.experimental._config.patch(backed_size_oblivious=True),
                 torch_export_patches(patch_transformers=True),