functools wrap

BerkeleyHCI · Mar 29, 2024 · 359c87a · 359c87a
1 parent 17405f0
commit 359c87a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/edg_core/HierarchyBlock.py b/edg_core/HierarchyBlock.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
-import itertools
-from functools import reduce
+from functools import reduce, wraps
 from typing import *
 
 import edgir
@@ -40,6 +39,7 @@ def init_in_parent(fn: InitType) -> InitType:
   import inspect
   from .Builder import builder
 
+  @wraps(fn)
   def wrapped(self: Block, *args_tup, **kwargs) -> Any:
     args = list(args_tup)
     builder_prev = builder.get_curr_context()