apply default qubit state prep api change (#55)

PennyLaneAI · Jan 26, 2021 · 5e5b7a7 · 5e5b7a7
1 parent 14b50f9
commit 5e5b7a7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pennylane_lightning/lightning_qubit.py b/pennylane_lightning/lightning_qubit.py
@@ -118,7 +118,12 @@ def apply(self, operations, rotations=None, **kwargs):
         for i, operation in enumerate(operations):  # State preparation is currently done in Python
             if isinstance(operation, (QubitStateVector, BasisState)):
                 if i == 0:
-                    self._apply_operation(operation)
+
+                    if isinstance(operation, QubitStateVector):
+                        self._apply_state_vector(operation.parameters[0], operation.wires)
+                    else:
+                        self._apply_basis_state(operation.parameters[0], operation.wires)
+
                     del operations[0]
                 else:
                     raise DeviceError(