Raise original exception when test hits codemod.TransformFailure

Otherwise it's very hard to tell why the test failed.
pytorch-labs · Mar 11, 2024 · 5cc186a · 5cc186a
1 parent 8846f5c
commit 5cc186a
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