Raise original exception when test hits codemod.TransformFailure (#28)

* Raise original exception when test hits codemod.TransformFailure Otherwise it's very hard to tell why the test failed.
pytorch-labs · Mar 11, 2024 · 890b5cc · 890b5cc
1 parent 8846f5c
commit 890b5cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_torchfix.py b/tests/test_torchfix.py
@@ -27,6 +27,8 @@ def _codemod_results(source_path):
     config = TorchCodemodConfig(select=list(GET_ALL_ERROR_CODES()))
     context = TorchCodemod(codemod.CodemodContext(filename=source_path), config)
     new_module = codemod.transform_module(context, code)
+    if isinstance(new_module, codemod.TransformFailure):
+        raise new_module.error
     return new_module.code