Removing the draw function

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())

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)