Open v0.3 API + debiasing

pennylane_ionq/api_client.py

@@ -112,12 +112,16 @@
         api_key (str): IonQ cloud platform API key
     """
 
-    USER_AGENT = "pennylane-ionq-api-client/0.1"
-    HOSTNAME = "api.ionq.co/v0.1"
+    USER_AGENT = "pennylane-ionq-api-client/0.3"
+    HOSTNAME = "api.ionq.co/v0.3"
     BASE_URL = "https://{}".format(HOSTNAME)
 
     def __init__(self, **kwargs):
-        self.AUTHENTICATION_TOKEN = os.getenv("IONQ_API_KEY") or kwargs.get("api_key", None)
+        self.AUTHENTICATION_TOKEN = (
+            kwargs.get("api_key", None)
+            or os.getenv("PENNYLANE_IONQ_API_KEY")
+            or os.getenv("IONQ_API_KEY")
+        )
         self.DEBUG = False
 
         if "IONQ_DEBUG" in os.environ:
@@ -196,17 +200,18 @@
 
         return response
 
-    def get(self, path):
+    def get(self, path, params=None):
         """
         Sends a GET request to the provided path. Returns a response object.
 
         Args:
             path (str): path to send the GET request to
+            params (dict): parameters to include in the request
 
         Returns:
             requests.Response: A response object, or None if no response could be fetched
         """
-        return self.request(requests.get, url=self.join_path(path))
+        return self.request(requests.get, url=self.join_path(path), params=params)
 
     def post(self, path, payload):
         """
@@ -220,7 +225,9 @@
         Returns:
             requests.Response: A response object, or None if no response could be fetched
         """
-        return self.request(requests.post, url=self.join_path(path), data=json.dumps(payload))
+        return self.request(
+            requests.post, url=self.join_path(path), data=json.dumps(payload)
+        )
 
 
 class ResourceManager:
@@ -249,7 +256,7 @@
         """
         return join_path(self.resource.PATH, path)
 
-    def get(self, resource_id=None):
+    def get(self, resource_id=None, params=None):
         """
         Attempts to retrieve a particular record by sending a GET
         request to the appropriate endpoint. If successful, the resource
@@ -259,12 +266,14 @@
             resource_id (int): the ID of an object to be retrieved
         """
         if "GET" not in self.resource.SUPPORTED_METHODS:
-            raise MethodNotSupportedException("GET method on this resource is not supported")
+            raise MethodNotSupportedException(
+                "GET method on this resource is not supported"
+            )
 
         if resource_id is not None:
-            response = self.client.get(self.join_path(str(resource_id)))
+            response = self.client.get(self.join_path(str(resource_id)), params=params)
         else:
-            response = self.client.get(self.resource.PATH)
+            response = self.client.get(self.resource.PATH, params=params)
         self.handle_response(response)
 
     def create(self, **params):
@@ -276,7 +285,9 @@
             **params: arbitrary parameters to be passed on to the POST request
         """
         if "POST" not in self.resource.SUPPORTED_METHODS:
-            raise MethodNotSupportedException("POST method on this resource is not supported")
+            raise MethodNotSupportedException(
+                "POST method on this resource is not supported"
+            )
 
         if self.resource.id:
             raise ObjectAlreadyCreatedException("ID must be None when calling create")
@@ -331,7 +342,7 @@
         self.errors.append(error)
         try:
             response.raise_for_status()
         except Exception as e:
             raise Exception(response.text) from e
 
     def refresh_data(self, data):

pennylane_ionq/device.py

@@ -93,14 +93,28 @@
     # and therefore does not support the Hermitian observable.
     observables = {"PauliX", "PauliY", "PauliZ", "Hadamard", "Identity"}
 
-    def __init__(self, wires, *, target="simulator", gateset="qis", shots=1024, api_key=None):
+    def __init__(
+        self,
+        wires,
+        *,
+        target="simulator",
+        gateset="qis",
+        shots=1024,
+        error_mitigation=None,
+        sharpen=None,
+        api_key=None,
+    ):
         if shots is None:
-            raise ValueError("The ionq device does not support analytic expectation values.")
+            raise ValueError(
+                "The ionq device does not support analytic expectation values."
+            )
 
         super().__init__(wires=wires, shots=shots)
         self.target = target
         self.api_key = api_key
         self.gateset = gateset
+        self.error_mitigation = error_mitigation
+        self.sharpen = sharpen
         self._operation_map = _GATESET_OPS[gateset]
         self.reset()
 
@@ -109,16 +123,25 @@
         self._prob_array = None
         self.histogram = None
         self.circuit = {
+            "format": "ionq.circuit.v0",
             "qubits": self.num_wires,
             "circuit": [],
             "gateset": self.gateset,
         }
         self.job = {
-            "lang": "json",
-            "body": self.circuit,
+            "input": self.circuit,
             "target": self.target,
             "shots": self.shots,
         }
+        if self.error_mitigation is not None:
+            self.job["error_mitigation"] = self.error_mitigation
+        if self.job["target"] == "qpu":
+            self.job["target"] = "qpu.harmony"
+            warnings.warn(
+                "The ionq_qpu backend is deprecated. Defaulting to ionq_qpu.harmony.",
+                DeprecationWarning,
+                stacklevel=2,
+            )
 
     @property
     def operations(self):
@@ -132,9 +155,12 @@
     def apply(self, operations, **kwargs):
         self.reset()
         rotations = kwargs.pop("rotations", [])
+        self.error_mitigation = kwargs.pop("error_mitigation", {})
 
         if len(operations) == 0 and len(rotations) == 0:
-            warnings.warn("Circuit is empty. Empty circuits return failures. Submitting anyway.")
+            warnings.warn(
+                "Circuit is empty. Empty circuits return failures. Submitting anyway."
+            )
 
         for i, operation in enumerate(operations):
             if i > 0 and operation.name in {"BasisState", "QubitStateVector"}:
@@ -188,7 +214,10 @@
             if job.is_failed:
                 raise JobExecutionError("Job failed")
 
-        job.manager.get(job.id.value)
+        if self.sharpen is not None:
+            params["sharpen"] = sharpen
+
+        job.manager.get(resource_id=job.id.value, params=params)
 
         # The returned job histogram is of the form
         # dict[str, float], and maps the computational basis
@@ -210,7 +239,8 @@
             # Here, we rearrange the states to match the big-endian ordering
             # expected by PennyLane.
             basis_states = (
-                int(bin(int(k))[2:].rjust(self.num_wires, "0")[::-1], 2) for k in self.histogram
+                int(bin(int(k))[2:].rjust(self.num_wires, "0")[::-1], 2)
+                for k in self.histogram
             )
             idx = np.fromiter(basis_states, dtype=int)
 
@@ -230,7 +260,9 @@
         if shot_range is None and bin_size is None:
             return self.marginal_prob(self.prob, wires)
 
-        return self.estimate_probability(wires=wires, shot_range=shot_range, bin_size=bin_size)
+        return self.estimate_probability(
+            wires=wires, shot_range=shot_range, bin_size=bin_size
+        )
 
 
 class SimulatorDevice(IonQDevice):
@@ -251,8 +283,12 @@
     name = "IonQ Simulator PennyLane plugin"
     short_name = "ionq.simulator"
 
-    def __init__(self, wires, *, target="simulator", gateset="qis", shots=1024, api_key=None):
-        super().__init__(wires=wires, target=target, gateset=gateset, shots=shots, api_key=api_key)
+    def __init__(
+        self, wires, *, target="simulator", gateset="qis", shots=1024, api_key=None
+    ):
+        super().__init__(
+            wires=wires, target=target, gateset=gateset, shots=shots, api_key=api_key
+        )
 
     def generate_samples(self):
         """Generates samples by random sampling with the probabilities returned by the simulator."""
@@ -279,8 +315,26 @@
     name = "IonQ QPU PennyLane plugin"
     short_name = "ionq.qpu"
 
-    def __init__(self, wires, *, target="qpu", gateset="qis", shots=1024, api_key=None):
-        super().__init__(wires=wires, target=target, gateset=gateset, shots=shots, api_key=api_key)
+    def __init__(
+        self,
+        wires,
+        *,
+        target="qpu",
+        gateset="qis",
+        shots=1024,
+        error_mitigation=None,
+        sharpen=None,
+        api_key=None,
+    ):
+        super().__init__(
+            wires=wires,
+            target=target,
+            gateset=gateset,
+            shots=shots,
+            error_mitigation=error_mitigation,
+            sharpen=sharpen,
+            api_key=api_key,
+        )
 
     def generate_samples(self):
         """Generates samples from the qpu.