fix: Update device capabilities list for Rigetti Ankaa-3 launch (#193)

Author: licedric

qiskit_braket_provider/providers/adapter.py

@@ -23,7 +23,10 @@
 from braket.device_schema.ionq import IonqDeviceCapabilities
 from braket.device_schema.iqm import IqmDeviceCapabilities
 from braket.device_schema.oqc import OqcDeviceCapabilities
-from braket.device_schema.rigetti import RigettiDeviceCapabilities
+from braket.device_schema.rigetti import (
+    RigettiDeviceCapabilities,
+    RigettiDeviceCapabilitiesV2,
+)
 from braket.device_schema.simulators import GateModelSimulatorDeviceCapabilities
 from braket.devices import LocalSimulator
 from braket.ir.openqasm.modifiers import Control
@@ -321,6 +324,7 @@ def aws_device_to_target(device: AwsDevice) -> Target:
         (
             IonqDeviceCapabilities,
             RigettiDeviceCapabilities,
+            RigettiDeviceCapabilitiesV2,
             OqcDeviceCapabilities,
             IqmDeviceCapabilities,
         ),
@@ -355,6 +359,7 @@ def _qpu_target(
     properties: Union[
         IonqDeviceCapabilities,
         RigettiDeviceCapabilities,
+        RigettiDeviceCapabilitiesV2,
         OqcDeviceCapabilities,
         IqmDeviceCapabilities,
     ],
@@ -400,7 +405,9 @@ def _2q_instruction_properties(qubit_count, connectivity, properties):
 
     # building coupling map for device with connectivity graph
     else:
-        if isinstance(properties, RigettiDeviceCapabilities):
+        if isinstance(
+            properties, (RigettiDeviceCapabilities, RigettiDeviceCapabilitiesV2)
+        ):
             connectivity.connectivityGraph = _convert_aspen_qubit_indices(
                 connectivity.connectivityGraph
             )

tests/providers/test_braket_quantum_task.py

@@ -48,9 +48,13 @@ def test_result(self):
         task = self._get_task()
 
         self.assertEqual(task.result().job_id, "AwesomeId")
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].data.counts, {"01": 1, "10": 2})
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].data.memory, ["10", "10", "01"])
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].status, "COMPLETED")
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].shots, 3)
         self.assertEqual(task.result().get_memory(), ["10", "10", "01"])
 
@@ -69,7 +73,7 @@ def test_task_result_is_none(self, mock_retrieve_results):
         )
         result = task.result()
 
-        assert result.results is None
+        assert result.results is None or result.results == [None]
         mock_retrieve_results.assert_called_once()
 
     @patch("qiskit_braket_provider.providers.braket_quantum_task.AwsQuantumTask")
@@ -144,9 +148,13 @@ def test_result(self):
         task = self._get_task()
 
         self.assertEqual(task.result().job_id, "AwesomeId")
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].data.counts, {"01": 1, "10": 2})
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].data.memory, ["10", "10", "01"])
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].status, "COMPLETED")
+        # pylint: disable-next=no-member
         self.assertEqual(task.result().results[0].shots, 3)
         self.assertEqual(task.result().get_memory(), ["10", "10", "01"])
 
@@ -177,9 +185,13 @@ def test_AWS_result(self):
         job = self._get_job()
 
         self.assertEqual(job.result().job_id, "AwesomeId")
+        # pylint: disable-next=no-member
         self.assertEqual(job.result().results[0].data.counts, {"01": 1, "10": 2})
+        # pylint: disable-next=no-member
         self.assertEqual(job.result().results[0].data.memory, ["10", "10", "01"])
+        # pylint: disable-next=no-member
         self.assertEqual(job.result().results[0].status, "COMPLETED")
+        # pylint: disable-next=no-member
         self.assertEqual(job.result().results[0].shots, 3)
         self.assertEqual(job.result().get_memory(), ["10", "10", "01"])