trying another wrap

Qiskit · Feb 17, 2025 · e51cf2c · e51cf2c
1 parent 111345e
commit e51cf2c
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/qiskit/dagcircuit/dagnode.py b/qiskit/dagcircuit/dagnode.py
@@ -35,25 +35,31 @@
     from qiskit.dagcircuit import DAGCircuit
 
 
-def __init__wrapper__(func):
-    def wrapper(*args, **kwargs):
+def __init__wrapper__(cls):
+    old_init = cls.__init__
+
+    def new_init(self, *_, **kwargs):
         if "dag" in kwargs:
             warnings.warn(
-                f"The 'dag' parameter in {func.__module__}.{func.__qualname__} "
+                f"The 'dag' parameter in {self.__class__.__qualname__} "
                 "constructor is unused and it will be removed in Qiskit 2.0.",
                 category=DeprecationWarning,
                 stacklevel=2,
             )
-        res = func(*args, **kwargs)
-        return res
+        old_init(self)
+
+    return new_init
 
-    return wrapper
 
+DAGNode = qiskit._accelerate.circuit.DAGNode
+DAGOpNode = qiskit._accelerate.circuit.DAGOpNode
+DAGInNode = qiskit._accelerate.circuit.DAGInNode
+DAGOutNode = qiskit._accelerate.circuit.DAGOutNode
 
-DAGNode = __init__wrapper__(qiskit._accelerate.circuit.DAGNode)
-DAGOpNode = __init__wrapper__(qiskit._accelerate.circuit.DAGOpNode)
-DAGInNode = __init__wrapper__(qiskit._accelerate.circuit.DAGInNode)
-DAGOutNode = __init__wrapper__(qiskit._accelerate.circuit.DAGOutNode)
+DAGNode.__init__ = __init__wrapper__(qiskit._accelerate.circuit.DAGNode)
+DAGOpNode.__init__ = __init__wrapper__(qiskit._accelerate.circuit.DAGOpNode)
+DAGInNode.__init__ = __init__wrapper__(qiskit._accelerate.circuit.DAGInNode)
+DAGOutNode.__init__ = __init__wrapper__(qiskit._accelerate.circuit.DAGOutNode)
 
 
 def _legacy_condition_eq(cond1, cond2, bit_indices1, bit_indices2) -> bool: