Fix indexing in code example

PennyLaneAI · Sep 17, 2024 · bc68858 · bc68858
1 parent 86e3dcb
commit bc68858
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pennylane/templates/subroutines/arbitrary_unitary.py b/pennylane/templates/subroutines/arbitrary_unitary.py
@@ -83,8 +83,8 @@ class ArbitraryUnitary(Operation):
  .. code-block:: python
 
  def arbitrary_nearest_neighbour_interaction(weights, wires):
- for w in range(0, len(wires) - 1, 2):
- ArbitraryUnitary(weights, wires=[w, w + 1])
+ for i, w in enumerate(range(0, len(wires) - 1, 2)):
+ ArbitraryUnitary(weights[i], wires=[w, w + 1])
 
  Args:
  weights (tensor_like): The angles of the Pauli word rotations, needs to have length :math:`4^n - 1`