code: def original_qfunc_for_grad(angles):
      qml.Hadamard(wires="a")
      qml.RZ(angles[0], wires="a")
      qml.RX(angles[1], wires="b")
      qml.CNOT(wires=["b", "a"])
      return qml.expval(qml.PauliX(wires="a"))
  
imports: from itertools import product
  import pytest
  from gate_data import I, Z, S, T, H, X
  from test_optimization.utils import check_matrix_equivalence
  import pennylane
  from pennylane import numpy
  from pennylane.wires import Wires
  from pennylane.transforms import unitary_to_rot
label_desc: Use image node_id 2 for calling a global function with example usage: original_qfunc_for_grad(angles) and returns: qml