Merge pull request #213 from Exabyte-io/update/SOF-7324

Update/SOF-7324 fix last 2 tests

Author: VsevolodX

examples/system/get_authentication_params.ipynb

@@ -43,21 +43,53 @@
    ]
   },
   {
+   "metadata": {},
    "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "cellView": "form"
-   },
-   "outputs": [],
    "source": [
     "# @title Authorization Form\n",
-    "USERNAME = \"YOUR_USERNAME\"  # @param {type:\"string\"}\n",
+    "USERNAME = \"YOUR_USERNAME\"  # @param {type:\"string\"}"
+   ],
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "source": [
+    "import sys\n",
     "\n",
+    "# Get username if not set in parameters\n",
+    "if USERNAME == \"YOUR_USERNAME\":\n",
+    "    if sys.platform == \"emscripten\":\n",
+    "        USERNAME = await input(\"Please enter username: \")\n",
+    "    else:\n",
+    "        USERNAME = input(\"Please enter username: \")\n",
+    "print(f\"Using username: {USERNAME}\")"
+   ],
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "source": [
+    "# Get password\n",
     "# avoid storing password in plaintext\n",
     "from getpass import getpass\n",
     "\n",
+    "if sys.platform == \"emscripten\":\n",
+    "    PASSWORD = await getpass(\"Please enter password: \")\n",
+    "else:\n",
+    "    PASSWORD = getpass(\"Please enter password: \")"
+   ],
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "source": [
     "import os\n",
-    "import sys\n",
     "\n",
     "if \"COLAB_JUPYTER_IP\" in os.environ:\n",
     "    !GIT_BRANCH=\"dev\"; export GIT_BRANCH; curl -s \"https://raw.githubusercontent.com/Exabyte-io/api-examples/${GIT_BRANCH}/scripts/env.sh\" | bash\n",
@@ -78,11 +110,10 @@
     "    await micropip.install(\"mat3ra-api-examples\", deps=False)\n",
     "    from utils.jupyterlite import install_packages\n",
     "\n",
-    "    await install_packages(\"api_examples\")\n",
-    "    PASSWORD = await getpass(\"Please enter password: \")\n",
-    "else:\n",
-    "    PASSWORD = getpass(\"Please enter password: \")"
-   ]
+    "    await install_packages(\"api_examples\")"
+   ],
+   "outputs": [],
+   "execution_count": null
   },
   {
    "attachments": {},
@@ -94,15 +125,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
    "metadata": {},
-   "outputs": [],
    "source": [
     "from utils.settings import HOST, PORT, VERSION, SECURE\n",
     "from utils.generic import display_JSON\n",
     "\n",
     "from exabyte_api_client.endpoints.login import LoginEndpoint"
-   ]
+   ],
+   "outputs": [],
+   "execution_count": null
   },
   {
    "attachments": {},
@@ -114,13 +145,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
    "metadata": {},
-   "outputs": [],
    "source": [
     "endpoint = LoginEndpoint(HOST, PORT, USERNAME, PASSWORD, VERSION, SECURE)\n",
     "auth_params = endpoint.login()"
-   ]
+   ],
+   "outputs": [],
+   "execution_count": null
   },
   {
    "attachments": {},
@@ -134,16 +165,16 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
    "metadata": {
     "pycharm": {
      "name": "#%%\n"
     }
    },
-   "outputs": [],
    "source": [
     "display_JSON(auth_params)"
-   ]
+   ],
+   "outputs": [],
+   "execution_count": null
   }
  ],
  "metadata": {

examples/workflow/qe_scf_calculation.ipynb

@@ -44,8 +44,6 @@
    "id": "62f59cb7",
    "metadata": {},
    "source": [
-    "from examples.workflow.get_workflows import QUERY\n",
-    "\n",
     "# @title Authorization Form\n",
     "ACCOUNT_ID = \"ACCOUNT_ID\"  # @param {type:\"string\"}\n",
     "AUTH_TOKEN = \"AUTH_TOKEN\"  # @param {type:\"string\"}\n",
@@ -374,7 +372,8 @@
     "len_energy = len(\"total energy\")\n",
     "for line in output_file.split(\"\\n\"):\n",
     "    if line.strip().lstrip(\"!\")[:len_energy] == \"total energy\":\n",
-    "        energy.append(float(line.split(\"=\")[1].rstrip(\"Ry\")))"
+    "        energy.append(float(line.split(\"=\")[1].rstrip(\"Ry\")))\n",
+    "print(\"Total energy (Ry):\", energy[-1])"
    ],
    "outputs": [],
    "execution_count": null