Propagate query_timeout_seconds option for direct-access jobs. (#171)

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 * Support direct-access mode on AQT devices (#164)
 * Support up to 2000 shots per circuit on cloud resources (#165)
+* Fix result timeout propagation in direct-access mode (#171)
 
 ## qiskit-aqt-provider v1.5.0
 

qiskit_aqt_provider/aqt_job.py

@@ -447,7 +447,9 @@ def result(self) -> Result:
             for circuit_index, circuit in enumerate(self.circuits):
                 api_circuit = self.api_submit_payload.payload.circuits[circuit_index]
                 job_id = self._backend.submit(api_circuit)
-                api_result = self._backend.result(job_id)
+                api_result = self._backend.result(
+                    job_id, timeout=self.options.query_timeout_seconds
+                )
 
                 if isinstance(api_result.payload, JobResultError):
                     break

qiskit_aqt_provider/aqt_resource.py

@@ -337,18 +337,19 @@ def submit(self, circuit: api_models.QuantumCircuit) -> UUID:
         resp.raise_for_status()
         return UUID(resp.json())
 
-    def result(self, job_id: UUID) -> api_models_direct.JobResult:
+    def result(self, job_id: UUID, *, timeout: Optional[float]) -> api_models_direct.JobResult:
         """Query the result of a specific job.
 
         Block until a result (success or error) is available.
 
         Args:
             job_id: unique identifier of the target job.
+            timeout: query timeout, in seconds. Disabled if `None`.
 
         Returns:
             Job result, as API payload.
         """
-        resp = self._http_client.get(f"/circuit/result/{job_id}")
+        resp = self._http_client.get(f"/circuit/result/{job_id}", timeout=timeout)
         resp.raise_for_status()
         return api_models_direct.JobResult.model_validate(resp.json())