Update pennylane/transforms/insert_ops.py

PennyLaneAI · May 13, 2024 · e5216a8 · e5216a8
1 parent 976388e
commit e5216a8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pennylane/transforms/insert_ops.py b/pennylane/transforms/insert_ops.py
@@ -99,6 +99,8 @@ def insert(
 
     .. code-block:: python3
 
+        from functools import partial
+
         dev = qml.device("default.mixed", wires=2)
 
         @partial(qml.transforms.insert, op=qml.AmplitudeDamping, op_args=0.2, position="end")