Client does not hang

jobs/.gitignore

@@ -0,0 +1,2 @@
+*
+!.gitignore

src/qibo_client/qibo_client.py

@@ -135,7 +135,7 @@
         check_response_has_keys(response, ["qibo_version"])
         qibo_server_version = response.json()["qibo_version"]
 
-        if qibo_local_version != qibo_server_version:
+        if qibo_local_version < qibo_server_version:
             logger.warning(
                 "Local Qibo package version does not match the server one, please "
                 "upgrade: %s -> %s",
@@ -144,7 +144,11 @@
             )
 
     def run_circuit(
-        self, circuit: qibo.Circuit, nshots: int = 1000, device: str = "sim"
+        self,
+        circuit: qibo.Circuit,
+        nshots: int = 1000,
+        device: str = "sim",
+        wait_for_results: bool = True,
     ) -> Optional[np.ndarray]:
         """Run circuit on the cluster.
 
@@ -154,6 +158,8 @@
         :type nshots: int
         :param device: the device to run the circuit on. Default device is `sim`
         :type device: str
+        :param wait_for_results: wheter to let the client hang until server results are ready or not. Defaults to True.
+        :type wait_for_results: bool
 
         :return:
             the numpy array with the results of the computation. None if the job
@@ -169,8 +175,17 @@
             logger.error(err.message)
             return None
 
-        # retrieve results
         logger.info("Job posted on server with pid %s", self.pid)
+
+        if not wait_for_results:
+            logger.info(
+                "Check results availability for %s job in your reserved page at "
+                "https://cloud.qibo.science",
+                self.pid,
+            )
+            return None
+
+        # retrieve results
         logger.info(
             "Check results every %d seconds ...", constants.SECONDS_BETWEEN_CHECKS
         )
@@ -198,6 +213,7 @@
 
         # checks
         response.raise_for_status()
+
         check_response_has_keys(response, ["pid", "message"])
 
         # save the response

tests/test_qibo_client.py

@@ -183,6 +183,19 @@ def _new_side_effect(url, json, timeout):
     assert return_value is None
 
 
+def test__run_circuit_without_waiting_for_results(mock_request: Mock):
+    def _new_side_effect(url, json, timeout):
+        json_data = {"pid": None, "message": "post job to queue failed"}
+        return utils.MockedResponse(status_code=200, json_data=json_data)
+
+    mock_request.post.side_effect = _new_side_effect
+
+    client = _get_tii_client()
+    return_value = client.run_circuit(utils.MockedCircuit(), wait_for_results=False)
+
+    assert return_value is None
+
+
 def test_wait_for_response_to_get_request(mock_request: Mock):
     failed_attempts = 3
     url = "http://example.url"