from braket.aws import AwsQuantumTask
from braket.aws.aws_quantum_task import _create_annealing_device_params
from braket.aws.aws_session import AwsSession
from braket.circuits import Circuit
from braket.device_schema import GateModelParameters
from braket.device_schema.dwave import (
Dwave2000QDeviceParameters,
DwaveAdvantageDeviceParameters,
DwaveDeviceParameters,
)
from braket.device_schema.ionq import IonqDeviceParameters
from braket.device_schema.rigetti import RigettiDeviceParameters
from braket.device_schema.simulators import GateModelSimulatorDeviceParameters
from braket.tasks import AnnealingQuantumTaskResult, GateModelQuantumTaskResult
S3_TARGET = AwsSession.S3DestinationFolder("foo", "bar")
@pytest.fixture
def aws_session():
mock = Mock()
_mock_metadata(mock, "RUNNING")
return mock
@pytest.fixture
def quantum_task(aws_session):
return AwsQuantumTask("foo:bar:arn", aws_session, poll_timeout_seconds=2)
@pytest.fixture
def s3_destination_folder(s3_bucket, s3_prefix):
return AwsSession.S3DestinationFolder(s3_bucket, s3_prefix)
