From fff2644a44212cf34c7443810a7db16eadd71748 Mon Sep 17 00:00:00 2001 From: Coull Date: Wed, 16 Oct 2024 11:15:31 -0700 Subject: [PATCH] add missing type --- src/braket/aws/aws_quantum_task_batch.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/braket/aws/aws_quantum_task_batch.py b/src/braket/aws/aws_quantum_task_batch.py index f4f1d328a..736de586d 100644 --- a/src/braket/aws/aws_quantum_task_batch.py +++ b/src/braket/aws/aws_quantum_task_batch.py @@ -337,7 +337,10 @@ def results( max_retries: int = MAX_RETRIES, use_cached_value: bool = True, ) -> list[ - GateModelQuantumTaskResult | AnnealingQuantumTaskResult | PhotonicModelQuantumTaskResult + GateModelQuantumTaskResult + | AnnealingQuantumTaskResult + | PhotonicModelQuantumTaskResult + | AnalogHamiltonianSimulationQuantumTaskResult ]: """Retrieves the result of every quantum task in the batch. @@ -355,7 +358,7 @@ def results( even when results have already been cached. Default: `True`. Returns: - list[GateModelQuantumTaskResult | AnnealingQuantumTaskResult | PhotonicModelQuantumTaskResult]: The # noqa: E501 + list[GateModelQuantumTaskResult | AnnealingQuantumTaskResult | PhotonicModelQuantumTaskResult | AnalogHamiltonianSimulationQuantumTaskResult]: The # noqa: E501 results of all of the quantum tasks in the batch. `FAILED`, `CANCELLED`, or timed out quantum tasks will have a result of None """ @@ -380,7 +383,10 @@ def results( def _retrieve_results( tasks: list[AwsQuantumTask], max_workers: int ) -> list[ - GateModelQuantumTaskResult | AnnealingQuantumTaskResult | PhotonicModelQuantumTaskResult + GateModelQuantumTaskResult + | AnnealingQuantumTaskResult + | PhotonicModelQuantumTaskResult + | AnalogHamiltonianSimulationQuantumTaskResult ]: with ThreadPoolExecutor(max_workers=max_workers) as executor: result_futures = [executor.submit(task.result) for task in tasks]