Persist circuits in batch execute

doc/devices/braket_remote.rst

@@ -66,6 +66,10 @@ You can set a timeout by using the ``poll_timeout_seconds`` argument;
 the device will retry circuits that do not complete within the timeout.
 A timeout of 30 to 60 seconds is recommended for circuits with fewer than 25 qubits.
 
+Each of the submitted circuit can be visualised using the attribute ``circuits`` on the device
+
+>> print(remote_device.circuits[0])
+
 Device options
 ~~~~~~~~~~~~~~
 

src/braket/pennylane_plugin/braket_device.py

@@ -139,6 +139,7 @@ def __init__(
         super().__init__(wires, shots=shots or None)
         self._device = device
         self._circuit = None
+        self._circuits = []
         self._task = None
         self._noise_model = noise_model
         self._parametrize_differentiable = parametrize_differentiable
@@ -153,6 +154,7 @@ def __init__(
     def reset(self):
         super().reset()
         self._circuit = None
+        self._circuits = []
         self._task = None
 
     @property
@@ -584,6 +586,7 @@ def __init__(
         self._max_parallel = max_parallel
         self._max_connections = max_connections
         self._max_retries = max_retries
+        self.circuits = []
 
     @property
     def use_grouping(self) -> bool:
@@ -621,6 +624,7 @@ def batch_execute(self, circuits, **run_kwargs):
                     **run_kwargs,
                 )
             )
+            self.circuits.append(circuit)
 
         batch_shots = 0 if self.analytic else self.shots
 

test/unit_tests/test_braket_device.py

@@ -910,6 +910,25 @@ def test_batch_execute_non_parallel_tracker(mock_run):
     callback.assert_called_with(latest=latest, history=history, totals=totals)
 
 
+@patch.object(AwsDevice, "run_batch")
+def test_batch_execute_parallel_circuits_persistance(mock_run_batch):
+    mock_run_batch.return_value = TASK_BATCH
+    dev = _aws_device(wires=4, foo="bar", parallel=True)
+    assert dev.parallel is True
+
+    with QuantumTape() as circuit:
+        qml.Hadamard(wires=0)
+        qml.CNOT(wires=[0, 1])
+        qml.probs(wires=[0])
+        qml.expval(qml.PauliX(1))
+        qml.var(qml.PauliY(2))
+        qml.sample(qml.PauliZ(3))
+
+    circuits = [circuit, circuit]
+    dev.batch_execute(circuits)
+    assert dev.circuits[1]
+
+
 @patch.object(AwsDevice, "run_batch")
 def test_batch_execute_parallel(mock_run_batch):
     """Test batch_execute(parallel=True) correctly calls batch execution methods in Braket SDK"""