[skip ci] Removing skips from more vjp tests

PennyLaneAI · Mar 19, 2024 · aeee6ae · aeee6ae
1 parent 2a7ff4f
commit aeee6ae
Show file tree

Hide file tree

Showing 2 changed files with 433 additions and 405 deletions.
diff --git a/pennylane_lightning/lightning_qubit/_adjoint_jacobian.py b/pennylane_lightning/lightning_qubit/_adjoint_jacobian.py
@@ -293,6 +293,12 @@ def calculate_vjp(self, tape: QuantumTape, grad_vec):
                 "Adjoint differentiation does not support State measurements."
             )
 
+        if any(m.return_type is not Expectation for m in tape.measurements):
+            raise QuantumFunctionError(
+                "Adjoint differentiation method does not support expectation return type "
+                "mixed with other return types"
+            )
+
         # Proceed, because tape_return_type is Expectation.
         if len(grad_vec) != len(measurements):
             raise ValueError(