# Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
import json
from typing import Dict, Optional, Union
from pydantic import Field
from braket.device_schema.device_action_properties import DeviceActionType
from braket.device_schema.device_capabilities import DeviceCapabilities
from braket.device_schema.gate_model_qpu_paradigm_properties_v1 import (
GateModelQpuParadigmProperties,
)
from braket.device_schema.ionq.ionq_provider_properties_v1 import IonqProviderProperties
from braket.device_schema.jaqcd_device_action_properties import JaqcdDeviceActionProperties
from braket.device_schema.openqasm_device_action_properties import OpenQASMDeviceActionProperties
from braket.schema_common import BraketSchemaBase, BraketSchemaHeader
def _loads_with_provider(serialized: str) -> dict:
deserialized = json.loads(serialized)
provider = deserialized.get("provider")
deserialized["provider"] = (
IonqProviderProperties.parse_raw(json.dumps(provider)).dict() if provider else None
)
return deserialized
def _dumps_with_provider(payload: dict, **kwargs):
provider = payload.get("provider")
payload["provider"] = (
json.loads(IonqProviderProperties.parse_obj(provider).json()) if provider else None
)
return json.dumps(payload, **kwargs)
[docs]class IonqDeviceCapabilities(BraketSchemaBase, DeviceCapabilities):
"""
This defines the capabilities of an IonQ device.
Attributes:
action(Dict[Union[DeviceActionType, str],
Union[OpenQASMDeviceActionProperties, JaqcdDeviceActionProperties]]): Actions that an
IonQ device can support
paradigm(GateModelQpuParadigmProperties): Paradigm properties
provider(Optional[IonqProviderProperties]): IonQ provider specific properties
Examples:
>>> import json
>>> input_json = {
... "braketSchemaHeader": {
... "name": "braket.device_schema.ionq.ionq_device_capabilities",
... "version": "1",
... },
... "service": {
... "braketSchemaHeader": {
... "name": "braket.device_schema.device_service_properties",
... "version": "1",
... },
... "executionWindows": [
... {
... "executionDay": "Everyday",
... "windowStartHour": "09:00",
... "windowEndHour": "09:05",
... }
... ],
... "shotsRange": [1, 10],
... "deviceCost": {
... "price": 0.25,
... "unit": "minute"
... },
... "deviceDocumentation": {
... "imageUrl": "image_url",
... "summary": "Summary on the device",
... "externalDocumentationUrl": "external doc link",
... },
... "deviceLocation": "us-east-1",
... "updatedAt": "2020-06-16T19:28:02.869136"
... },
... "action": {
... "braket.ir.jaqcd.program": {
... "actionType": "braket.ir.jaqcd.program",
... "version": ["1"],
... "supportedOperations": ["x", "y"],
... "supportedResultTypes": [{
... "name": "resultType1",
... "observables": ["observable1"],
... "minShots": 0,
... "maxShots": 4,
... }],
... }
... },
... "paradigm": {
... "braketSchemaHeader": {
... "name": "braket.device_schema.gate_model_qpu_paradigm_properties",
... "version": "1",
... },
... "qubitCount": 11,
... "nativeGateSet": ["ccnot", "cy"],
... "connectivity": {
... "fullyConnected": False,
... "connectivityGraph": {"1": ["2", "3"]},
... },
... },
... "deviceParameters": {IonqDeviceParameters.schema_json()},
... }
>>> IonqDeviceCapabilities.parse_raw_schema(json.dumps(input_json))
"""
_PROGRAM_HEADER = BraketSchemaHeader(
name="braket.device_schema.ionq.ionq_device_capabilities", version="1"
)
braketSchemaHeader: BraketSchemaHeader = Field(default=_PROGRAM_HEADER, const=_PROGRAM_HEADER)
action: Dict[
Union[DeviceActionType, str],
Union[OpenQASMDeviceActionProperties, JaqcdDeviceActionProperties],
]
paradigm: GateModelQpuParadigmProperties
provider: Optional[IonqProviderProperties]
[docs] class Config:
# Pydantic does not use the custom encoders/decoders of nested models:
# https://github.com/pydantic/pydantic/issues/2277#issuecomment-1236369282
# This should be fixed in Pydantic v2:
# https://github.com/pydantic/pydantic/discussions/4456
json_loads = _loads_with_provider
json_dumps = _dumps_with_provider