revert batch_obs False

PennyLaneAI · Sep 9, 2024 · fbba326 · fbba326
1 parent a1cac8e
commit fbba326
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 34 deletions.
diff --git a/pennylane_lightning/lightning_kokkos/lightning_kokkos.py b/pennylane_lightning/lightning_kokkos/lightning_kokkos.py
@@ -307,11 +307,6 @@ def __init__(  # pylint: disable=too-many-arguments
                 "https://docs.pennylane.ai/projects/lightning/en/stable/dev/installation.html."
             )
 
-        if batch_obs:
-            raise DeviceError(
-                "Lightning Kokkos does not support batch observables option. 'batch_obs' should be False"
-            )
-
         super().__init__(
             wires=wires,
             c_dtype=c_dtype,

diff --git a/tests/test_comparison.py b/tests/test_comparison.py
@@ -78,12 +78,6 @@ def test_one_qubit_circuit(
     ):
         """Test a single-qubit circuit"""
 
-        if (
-            lightning_dev_version == lightning_backend_batch_obs_dev
-            and device_name == "lightning.kokkos"
-        ):
-            pytest.skip(reason="Lightning Kokkos does not support batch_obs")
-
         monkeypatch.setenv("OMP_NUM_THREADS", str(num_threads))
 
         def circuit(measurement):
@@ -124,12 +118,6 @@ def test_two_qubit_circuit(
     ):
         """Test a two-qubit circuit"""
 
-        if (
-            lightning_dev_version == lightning_backend_batch_obs_dev
-            and device_name == "lightning.kokkos"
-        ):
-            pytest.skip(reason="Lightning Kokkos does not support batch_obs")
-
         monkeypatch.setenv("OMP_NUM_THREADS", str(num_threads))
 
         def circuit(measurement):
@@ -179,12 +167,6 @@ def test_three_qubit_circuit(
     ):
         """Test a three-qubit circuit"""
 
-        if (
-            lightning_dev_version == lightning_backend_batch_obs_dev
-            and device_name == "lightning.kokkos"
-        ):
-            pytest.skip(reason="Lightning Kokkos does not support batch_obs")
-
         monkeypatch.setenv("OMP_NUM_THREADS", str(num_threads))
 
         def circuit(measurement):
@@ -242,12 +224,6 @@ def test_four_qubit_circuit(
     ):
         """Test a four-qubit circuit"""
 
-        if (
-            lightning_dev_version == lightning_backend_batch_obs_dev
-            and device_name == "lightning.kokkos"
-        ):
-            pytest.skip(reason="Lightning Kokkos does not support batch_obs")
-
         monkeypatch.setenv("OMP_NUM_THREADS", str(num_threads))
 
         def circuit(measurement):
@@ -308,11 +284,6 @@ def test_n_qubit_circuit(
         self, monkeypatch, stateprep, wires, lightning_dev_version, num_threads
     ):
         """Test an n-qubit circuit"""
-        if (
-            lightning_dev_version == lightning_backend_batch_obs_dev
-            and device_name == "lightning.kokkos"
-        ):
-            pytest.skip(reason="Lightning Kokkos does not support batch_obs")
 
         monkeypatch.setenv("OMP_NUM_THREADS", str(num_threads))