Fix websocket connection drop (#1090)

qiskit_ibm/api/clients/base.py

@@ -175,7 +175,7 @@ def stream(
                 logger.debug('Starting new websocket connection: %s using proxy %s',
                              url, self._proxy_params)
                 self._reset_state()
-                self._ws.run_forever(**self._proxy_params)
+                self._ws.run_forever(ping_interval=60, ping_timeout=10, **self._proxy_params)
                 self.connected = False
 
                 logger.debug("Websocket run_forever finished.")

releasenotes/notes/fix-websocket-connection-drop-3892e313554e99df.yaml

@@ -0,0 +1,6 @@
+---
+fixes:
+  - |
+    Fixes `#209 <https://github.com/Qiskit-Partners/qiskit-ibm/issues/209>`__ where the websocket
+    connection kept timing out when streaming results for a runtime job, due to inactivity,
+    when the job is in a pending state for a long time.

test/ibm/runtime/test_runtime_integration.py

@@ -506,8 +506,6 @@ def result_callback(job_id, interim_result):
         callback_err = []
         iterations = 3
         job = self._run_program(iterations=iterations, interim_results=int_res)
-
-        self._wait_for_status(job, JobStatus.RUNNING)
         job.stream_results(result_callback)
         job.wait_for_final_state()
         self.assertEqual(iterations-1, final_it)