Update quantum_run_stream to not prefetch first result (quantumlib#5024)

Review: @wcourtney
95-martin-orion · Mar 2, 2022 · 9fb7c1a · 9fb7c1a
1 parent 72f140b
commit 9fb7c1a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...-google/cirq_google/engine/client/quantum_v1alpha1/gapic/quantum_engine_service_client.py b/...-google/cirq_google/engine/client/quantum_v1alpha1/gapic/quantum_engine_service_client.py
@@ -2106,6 +2106,9 @@ def quantum_run_stream(
                 default_timeout=self._method_configs['QuantumRunStream'].timeout,
                 client_info=self._client_info,
             )
+            # Don't prefetch first result from stream, since this will cause deadlocks.
+            # See https://github.com/googleapis/python-api-core/pull/30
+            self.transport.quantum_run_stream._prefetch_first_result_ = False
 
         return self._inner_api_calls['quantum_run_stream'](
             requests, retry=retry, timeout=timeout, metadata=metadata