WIP implementing IQM device, as well as unifying all common devices' logics into the base scaleway device for cleaner and quicker integration.

Author: smaugdela

examples/hello_world.py

@@ -14,6 +14,7 @@
     secret_key=secret_key,
     backend=backend,
 ) as dev:
+
     @qml.set_shots(100)
     @qml.qnode(dev)
     def bell_state_circuit():

pennylane_scaleway/aer_device.py

@@ -14,22 +14,16 @@
 import numpy as np
 import warnings
 
-from dataclasses import replace, fields
+from dataclasses import fields
 from tenacity import retry, stop_after_attempt, stop_after_delay
 from typing import Callable, Iterable, List, Sequence, Tuple
 
 import pennylane as qml
 from pennylane.devices import ExecutionConfig
 from pennylane.devices.modifiers import simulator_tracking, single_tape_support
-from pennylane.devices.preprocess import (
-    decompose,
-    validate_device_wires,
-    validate_measurements,
-    validate_observables,
-)
 from pennylane.measurements import ExpectationMP, VarianceMP, MeasurementProcess
 from pennylane.tape import QuantumScript, QuantumScriptOrBatch
-from pennylane.transforms import split_non_commuting, broadcast_expand, transform
+from pennylane.transforms import transform
 from pennylane.transforms.core import TransformProgram
 
 from qiskit.primitives.containers import PrimitiveResult, PubResult
@@ -43,10 +37,8 @@
 from pennylane_scaleway.scw_device import ScalewayDevice
 from pennylane_scaleway.utils import (
     QISKIT_OPERATION_MAP,
-    accepted_sample_measurement,
     circuit_to_qiskit,
 )
-from pennylane_scaleway.utils import analytic_warning
 
 
 @simulator_tracking  # update device.tracker with some relevant information
@@ -161,35 +153,8 @@ def preprocess(
         self,
         execution_config: ExecutionConfig | None = None,
     ) -> tuple[TransformProgram, ExecutionConfig]:
-        config = execution_config or ExecutionConfig()
-        config = replace(config, use_device_gradient=False)
-
-        transform_program = TransformProgram()
-
-        transform_program.add_transform(analytic_warning)
-        transform_program.add_transform(
-            validate_device_wires, self.wires, name=self.name
-        )
-        transform_program.add_transform(
-            decompose,
-            stopping_condition=lambda op: op.name in self.operations,
-            name=self.name,
-            skip_initial_state_prep=False,
-        )
-        transform_program.add_transform(
-            validate_measurements,
-            sample_measurements=accepted_sample_measurement,
-            name=self.name,
-        )
-        transform_program.add_transform(
-            validate_observables,
-            stopping_condition=lambda obs: obs.name in self.observables,
-            name=self.name,
-        )
-        transform_program.add_transform(broadcast_expand)
-        transform_program.add_transform(split_non_commuting)
+        transform_program, config = super().preprocess(execution_config)
         transform_program.add_transform(split_execution_types)
-
         return transform_program, config
 
     def execute(
@@ -199,7 +164,9 @@ def execute(
     ) -> List:
 
         if not self._session_id:
-            raise RuntimeError("No active session. Please instanciate the device using a context manager, or call start() first. You can also attach to an existing deduplication_id.")
+            raise RuntimeError(
+                "No active session. Please instanciate the device using a context manager, or call start() first. You can also attach to an existing deduplication_id."
+            )
 
         if isinstance(circuits, QuantumScript):
             circuits = [circuits]

pennylane_scaleway/aqt_device.py

@@ -14,22 +14,15 @@
 import numpy as np
 import warnings
 
-from dataclasses import replace
 from inspect import signature
 from typing import Callable, List, Sequence, Tuple, Union
 from tenacity import retry, stop_after_attempt, stop_after_delay
 
-import pennylane as qml
 from pennylane.devices import ExecutionConfig
 from pennylane.devices.modifiers import simulator_tracking, single_tape_support
-from pennylane.devices.preprocess import (
-    decompose,
-    validate_device_wires,
-    validate_measurements,
-    validate_observables,
-)
+
 from pennylane.tape import QuantumScriptOrBatch, QuantumScript, QuantumTape
-from pennylane.transforms import broadcast_expand, split_non_commuting, transform
+from pennylane.transforms import transform
 from pennylane.transforms.core import TransformProgram
 
 from qiskit.result import Result
@@ -38,8 +31,6 @@
 
 from pennylane_scaleway.utils import (
     circuit_to_qiskit,
-    accepted_sample_measurement,
-    analytic_warning,
 )
 from pennylane_scaleway.scw_device import ScalewayDevice
 
@@ -128,7 +119,7 @@ class AqtDevice(ScalewayDevice):
         "Hadamard",
     }
 
-    def __init__(self, shots=None, seed=None, **kwargs):
+    def __init__(self, wires=None, shots=None, seed=None, **kwargs):
         """
         Params:
 
@@ -164,7 +155,7 @@ def circuit():
             ```
         """
 
-        super().__init__(wires=12, kwargs=kwargs, shots=shots, seed=seed)
+        super().__init__(wires=wires, kwargs=kwargs, shots=shots, seed=seed)
 
         self._default_shots = None
         if isinstance(shots, int):
@@ -183,9 +174,6 @@ def _handle_kwargs(self, **kwargs):
         self._run_options.update(
             {
                 "session_name": self._session_options.get("name"),
-                "session_deduplication_id": self._session_options.get(
-                    "deduplication_id"
-                ),
                 "session_max_duration": self._session_options.get("max_duration"),
                 "session_max_idle_duration": self._session_options.get(
                     "max_idle_duration"
@@ -203,37 +191,10 @@ def preprocess(
         self,
         execution_config: ExecutionConfig | None = None,
     ) -> tuple[TransformProgram, ExecutionConfig]:
-        transform_program = TransformProgram()
-        config = execution_config or ExecutionConfig()
-        config = replace(config, use_device_gradient=False)
-
-        transform_program.add_transform(analytic_warning)
+        transform_program, config = super().preprocess(execution_config)
         transform_program.add_transform(
             limit_aqt_shots, default_shots=self._default_shots
         )
-        transform_program.add_transform(
-            validate_device_wires, self.wires, name=self.name
-        )
-        transform_program.add_transform(
-            decompose,
-            stopping_condition=lambda x: x.name in self.operations,
-            name=self.name,
-            skip_initial_state_prep=False,
-        )
-        transform_program.add_transform(
-            validate_measurements,
-            sample_measurements=accepted_sample_measurement,
-            name=self.name,
-        )
-        transform_program.add_transform(
-            validate_observables,
-            stopping_condition=lambda x: x.name in self.observables,
-            name=self.name,
-        )
-
-        transform_program.add_transform(broadcast_expand)
-        transform_program.add_transform(split_non_commuting)
-
         return transform_program, config
 
     def execute(
@@ -242,7 +203,9 @@ def execute(
         execution_config: ExecutionConfig | None = None,
     ) -> List:
         if not self._session_id:
-            raise RuntimeError("No active session. Please instanciate the device using a context manager, or call start() first. You can also attach to an existing deduplication_id.")
+            raise RuntimeError(
+                "No active session. Please instanciate the device using a context manager, or call start() first. You can also attach to an existing deduplication_id."
+            )
 
         if isinstance(circuits, QuantumScript):
             circuits = [circuits]