add write file

thuml · Nov 29, 2023 · 8800a13 · 8800a13
1 parent aa03288
commit 8800a13
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/depyf/explain/enable_debugging.py b/depyf/explain/enable_debugging.py
@@ -49,6 +49,8 @@ def __call__(self, code, new_code):
                     else:
                         func_name = filename.split(os.path.sep)[-1].split(".")[0]
                         src = f"def {func_name}" + src_body
+                        with open(filename, "w") as f:
+                            f.write(src)
                         break
 
             transformed_code = compile(src, filename=filename, mode="exec")