Merge pull request #40 from thuml/fix_attr

deal with none in patch
thuml · Jul 4, 2024 · 2633083 · 2633083
2 parents 7824657 + fc7d553
commit 2633083
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/depyf/VERSION.txt b/depyf/VERSION.txt
@@ -1 +1 @@
-0.15.0
+0.16.0
diff --git a/depyf/explain/enable_debugging.py b/depyf/explain/enable_debugging.py
@@ -122,11 +122,13 @@ def __call__(self, code, new_code):
 @contextlib.contextmanager
 def patch(parent, name, value):
     old_value = getattr(parent, name, None)
-    setattr(parent, name, value)
+    if old_value is not None:
+        setattr(parent, name, value)
     try:
         yield
     finally:
-        setattr(parent, name, old_value)
+        if old_value is not None:
+            setattr(parent, name, old_value)
 
 
 @contextlib.contextmanager