def test_from_annealing(device_parameters, aws_session, arn, problem):
mocked_task_arn = "task-arn-1"
aws_session.create_quantum_task.return_value = mocked_task_arn
task = AwsQuantumTask.create(
aws_session, arn, problem, S3_TARGET, 1000, device_parameters=device_parameters
)
assert task == AwsQuantumTask(
mocked_task_arn, aws_session, AnnealingQuantumTaskResult.from_string
)
_assert_create_quantum_task_called_with(
aws_session,
arn,
problem,
S3_TARGET,
1000,
DwaveDeviceParameters.parse_obj(device_parameters),
)
def _(
problem: Problem,
aws_session: AwsSession,
create_task_kwargs: Dict[str, Any],
device_parameters: Union[dict, DwaveDeviceParameters],
device_arn: str,
*args,
**kwargs,
) -> AwsQuantumTask:
create_task_kwargs.update({
"action":
problem.to_ir().json(),
"deviceParameters":
DwaveDeviceParameters.parse_obj(device_parameters).json(),
})
task_arn = aws_session.create_quantum_task(**create_task_kwargs)
return AwsQuantumTask(task_arn, aws_session, *args, **kwargs)
},
"arn:aws:braket:::device/qpu/d-wave/fake_arn",
marks=pytest.mark.xfail(reason="Bad ARN"),
),
(
{
"deviceLevelParameters": {
"postprocessingType": "OPTIMIZATION"
}
},
"arn:aws:braket:::device/qpu/d-wave/DW_2000Q_6",
),
(
DwaveDeviceParameters.parse_obj({
"providerLevelParameters": {
"postprocessingType": "OPTIMIZATION"
}
}),
"arn:aws:braket:::device/qpu/d-wave/Advantage_system1",
),
(
DwaveDeviceParameters.parse_obj({
"deviceLevelParameters": {
"postprocessingType": "OPTIMIZATION"
}
}),
"arn:aws:braket:::device/qpu/d-wave/Advantage_system1",
),
(
DwaveAdvantageDeviceParameters.parse_obj(
{"deviceLevelParameters": {
}
},
"arn:aws:braket:::device/qpu/d-wave/DW_2000Q_6",
),
(
DwaveDeviceParameters.parse_obj({
"providerLevelParameters": {
"postprocessingType": "OPTIMIZATION",
"annealingOffsets": [3.67, 6.123],
"annealingSchedule": [[13.37, 10.08], [3.14, 1.618]],
"annealingDuration": 1,
"autoScale": False,
"beta": 0.2,
"chains": [[0, 1, 5], [6]],
"compensateFluxDrift": False,
"fluxBiases": [1.1, 2.2, 3.3, 4.4],
"initialState": [1, 3, 0, 1],
"maxResults": 1,
"programmingThermalizationDuration": 625,
"readoutThermalizationDuration": 256,
"reduceIntersampleCorrelation": False,
"reinitializeState": True,
"resultFormat": "RAW",
"spinReversalTransformCount": 100,
}
}),
"arn:aws:braket:::device/qpu/d-wave/Advantage_system1",
),
(
DwaveDeviceParameters.parse_obj(
{
