fix: wait for jobs to finish async

Author: VsevolodX

examples/job/get-file-from-job.ipynb

@@ -140,7 +140,7 @@
     "# Import settings file and utils file\n",
     "from utils.settings import ENDPOINT_ARGS, ACCOUNT_ID\n",
     "from utils.api import (\n",
-    "    wait_for_jobs_to_finish,\n",
+    "    wait_for_jobs_to_finish_async,\n",
     "    get_property_by_subworkflow_and_unit_indicies,\n",
     ")\n",
     "from utils.visualize import (\n",
@@ -218,7 +218,7 @@
     "\n",
     "# Submit the job\n",
     "job_endpoints.submit(job[\"_id\"])\n",
-    "wait_for_jobs_to_finish(job_endpoints, [job[\"_id\"]])"
+    "await wait_for_jobs_to_finish_async(job_endpoints, [job[\"_id\"]])"
    ]
   },
   {

examples/job/ml-train-model-predict-properties.ipynb

@@ -142,7 +142,7 @@
     "from utils.settings import ENDPOINT_ARGS, ACCOUNT_ID\n",
     "from utils.api import (\n",
     "    copy_bank_workflow_by_system_name,\n",
-    "    wait_for_jobs_to_finish,\n",
+    "    wait_for_jobs_to_finish_async,\n",
     "    get_property_by_subworkflow_and_unit_indicies,\n",
     ")\n",
     "from utils.visualize import (\n",
@@ -434,7 +434,7 @@
    "outputs": [],
    "source": [
     "job_ids = [job[\"_id\"] for job in jobs]\n",
-    "wait_for_jobs_to_finish(job_endpoints, job_ids)"
+    "await wait_for_jobs_to_finish_async(job_endpoints, job_ids)"
    ]
   },
   {
@@ -505,9 +505,7 @@
     "outputId": "dfca0a46-6318-40a9-84a0-5e8925f5958e"
    },
    "outputs": [],
-   "source": [
-    "wait_for_jobs_to_finish(job_endpoints, [job[\"_id\"]])"
-   ]
+   "source": "await wait_for_jobs_to_finish_async(job_endpoints, [job[\"_id\"]])"
   },
   {
    "attachments": {},
@@ -628,9 +626,7 @@
     "outputId": "3892222c-1f00-4d83-d5b7-83195af3ec5a"
    },
    "outputs": [],
-   "source": [
-    "wait_for_jobs_to_finish(job_endpoints, [job[\"_id\"]])"
-   ]
+   "source": "await wait_for_jobs_to_finish_async(job_endpoints, [job[\"_id\"]])"
   },
   {
    "attachments": {},

examples/job/run-simulations-and-extract-properties.ipynb

@@ -150,7 +150,7 @@
     "\n",
     "# Import settings file and utils file\n",
     "from utils.settings import ENDPOINT_ARGS, ACCOUNT_ID\n",
-    "from utils.api import wait_for_jobs_to_finish, get_property_by_subworkflow_and_unit_indicies\n",
+    "from utils.api import wait_for_jobs_to_finish_async, get_property_by_subworkflow_and_unit_indicies\n",
     "from utils.visualize import dataframe_to_html\n",
     "\n",
     "import pandas as pd\n",
@@ -523,7 +523,7 @@
    "outputs": [],
    "source": [
     "job_ids = [job[\"_id\"] for job in jobs]\n",
-    "wait_for_jobs_to_finish(job_endpoints, job_ids)"
+    "await wait_for_jobs_to_finish_async(job_endpoints, job_ids)"
    ]
   },
   {

examples/reproducing_publications/band_gaps_for_interface_bilayer_twisted_molybdenum_disulfide.ipynb

@@ -763,7 +763,7 @@
     "\n",
     "job_ids = [job[\"_id\"] for job in jobs]\n",
     "\n",
-    "wait_for_jobs_to_finish(job_endpoints, job_ids, poll_interval=60)"
+    "await wait_for_jobs_to_finish_async(job_endpoints, job_ids, poll_interval=60)"
    ]
   },
   {

examples/reproducing_publications/band_structure_for_interface_bilayer_twisted_molybdenum_disulfide.ipynb

@@ -750,7 +750,7 @@
     "\n",
     "job_ids = [job[\"_id\"] for job in jobs]\n",
     "\n",
-    "wait_for_jobs_to_finish(job_endpoints, job_ids, poll_interval=60)"
+    "await wait_for_jobs_to_finish_async(job_endpoints, job_ids, poll_interval=60)"
    ]
   },
   {

examples/workflow/qe_scf_calculation.ipynb

@@ -337,7 +337,7 @@
    "outputs": [],
    "source": [
     "# monitor job and wait for it to be finished\n",
-    "wait_for_jobs_to_finish(job_endpoints, [JOB_RESP[\"_id\"]], 30)"
+    "await wait_for_jobs_to_finish_async(job_endpoints, [JOB_RESP[\"_id\"]], 30)"
    ]
   },
   {

other/jarvis/run_job_using_material_from_jarvis_db.ipynb

@@ -247,7 +247,7 @@
    "outputs": [],
    "source": [
     "from utils.settings import ENDPOINT_ARGS, ACCOUNT_ID\n",
-    "from utils.api import wait_for_jobs_to_finish, get_property_by_subworkflow_and_unit_indicies\n",
+    "from utils.api import wait_for_jobs_to_finish_async, get_property_by_subworkflow_and_unit_indicies\n",
     "from utils.visualize import dataframe_to_html, display_JSON\n",
     "\n",
     "# Relevant functions from the API client\n",
@@ -439,9 +439,9 @@
    },
    "outputs": [],
    "source": [
-    "from utils.api import wait_for_jobs_to_finish, get_property_by_subworkflow_and_unit_indicies\n",
+    "from utils.api import wait_for_jobs_to_finish_async, get_property_by_subworkflow_and_unit_indicies\n",
     "job_id = job[\"_id\"]\n",
-    "wait_for_jobs_to_finish(job_endpoints, [job_id])"
+    "await wait_for_jobs_to_finish_async(job_endpoints, [job_id])"
    ]
   },
   {