diff --git a/pennylane_lightning/lightning_tensor/quimb/_mps.py b/pennylane_lightning/lightning_tensor/quimb/_mps.py index d41e9b7ad0..9acb06933b 100644 --- a/pennylane_lightning/lightning_tensor/quimb/_mps.py +++ b/pennylane_lightning/lightning_tensor/quimb/_mps.py @@ -227,14 +227,6 @@ def state_to_array(self, digits: int = 5) -> np.ndarray: """Contract the MPS into a dense array and round the values.""" return self._circuitMPS.to_dense().round(digits) - def draw_state(self) -> None: - """Draw the MPS.""" - self._circuitMPS.psi.draw( - color=[f"I{q}" for q in range(len(self._wires))], - show_tags=False, - show_inds=True, - ) - def _reset_state(self) -> None: """Reset the MPS.""" self._circuitMPS = qtn.CircuitMPS(psi0=self._initial_mps()) diff --git a/tests/lightning_tensor/test_quimb_mps.py b/tests/lightning_tensor/test_quimb_mps.py index db3dd390a9..bdf4775bbd 100644 --- a/tests/lightning_tensor/test_quimb_mps.py +++ b/tests/lightning_tensor/test_quimb_mps.py @@ -205,3 +205,18 @@ def test_not_implemented_meas(self, backend, method): with pytest.raises(NotImplementedError): dev.execute(tape) + + def test_not_implemented_shots(self, backend, method): + """Tests that this interface does not support measurements with finite shots.""" + + ops = [qml.Identity(0)] + measurements = [qml.expval(qml.PauliZ(0))] + tape = qml.tape.QuantumScript(ops, measurements) + tape._shots = 5 + + dev = LightningTensor( + wires=tape.wires, backend=backend, method=method, c_dtype=np.complex64 + ) + + with pytest.raises(NotImplementedError): + dev.execute(tape)