diff --git a/pennylane_lightning/lightning_tensor/lightning_tensor.py b/pennylane_lightning/lightning_tensor/lightning_tensor.py index 03b5f2a92b..9916a4317d 100644 --- a/pennylane_lightning/lightning_tensor/lightning_tensor.py +++ b/pennylane_lightning/lightning_tensor/lightning_tensor.py @@ -66,9 +66,7 @@ class LightningTensor(Device): statistics like expectation values and variances is performed analytically. c_dtype: Datatypes for the tensor representation. Must be one of ``np.complex64`` or ``np.complex128``. - **kwargs: keyword arguments. The following options are supported: - - # TODO add when cuTensorNet MPS backend is available as a prototype + **kwargs: keyword arguments. TODO add when cuTensorNet MPS backend is available as a prototype. """ # pylint: disable=too-many-instance-attributes @@ -105,6 +103,8 @@ def __init__( self._method = method self._c_dtype = c_dtype + self._interface = None + for arg in kwargs: if arg not in self._device_options: raise TypeError( diff --git a/tests/lightning_tensor/test_lightning_tensor.py b/tests/lightning_tensor/test_lightning_tensor.py index aacc94bad3..4bb2c2d7fd 100644 --- a/tests/lightning_tensor/test_lightning_tensor.py +++ b/tests/lightning_tensor/test_lightning_tensor.py @@ -68,16 +68,13 @@ def test_invalid_method(method): def test_invalid_keyword_arg(): """Test an invalid keyword argument.""" - with pytest.raises( - TypeError, - match=f"Unexpected argument: fake_arg during initialization of the lightning.tensor device.", - ): + with pytest.raises(TypeError): LightningTensor(fake_arg=None) def test_invalid_shots(): """Test that an error is raised if finite number of shots are requestd.""" - with pytest.raises(ValueError, match="lightning.tensor does not support finite shots."): + with pytest.raises(ValueError): LightningTensor(shots=5)