fix nbs again gain (#157)

Author: makslevental

examples/cuda_e2e.ipynb

@@ -34,34 +34,29 @@
    "source": [
     "import os\n",
     "BRANCH = os.getenv(\"BRANCH\", \"main\")\n",
-    "os.environ[\"SCRIPT_ADDRESS\"] = f\"https://raw.githubusercontent.com/makslevental/mlir-python-extras/refs/heads/{BRANCH}/scripts/get_latest_bindings.py\"\n",
-    "!curl $SCRIPT_ADDRESS -o get_latest_bindings.py\n",
-    "!latest_cuda_version=$(python get_latest_bindings.py \"cuda\") && pip install -q mlir_python_bindings==$latest_cuda_version -f https://makslevental.github.io/wheels\n",
-    "!pip install -q git+https://github.com/makslevental/mlir-python-extras@$BRANCH"
+    "os.environ[\"SCRIPT_ADDRESS\"] = f\"https://raw.githubusercontent.com/makslevental/mlir-python-extras/refs/heads/{BRANCH}/scripts/get_latest_bindings.py\""
    ],
    "metadata": {
     "colab": {
      "base_uri": "https://localhost:8080/"
     },
     "id": "Xh-QUDWiX-FD",
-    "outputId": "6865a63a-daa4-4610-e33a-721d37c0211f",
-    "ExecuteTime": {
-     "end_time": "2025-05-15T22:44:10.812357Z",
-     "start_time": "2025-05-15T22:44:04.735503Z"
-    }
+    "outputId": "6865a63a-daa4-4610-e33a-721d37c0211f"
    },
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "  % Total    % Received % Xferd  Average Speed   Time    Time     Time  Current\r\n",
-      "                                 Dload  Upload   Total   Spent    Left  Speed\r\n",
-      "100  2421  100  2421    0     0  40214      0 --:--:-- --:--:-- --:--:-- 40350\r\n"
-     ]
-    }
-   ],
-   "execution_count": 9
+   "outputs": [],
+   "execution_count": null
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": [
+    "%%bash\n",
+    "curl $SCRIPT_ADDRESS -o get_latest_bindings.py\n",
+    "latest_cuda_version=$(python get_latest_bindings.py \"cuda\") && pip install -q mlir_python_bindings==$latest_cuda_version -f https://makslevental.github.io/wheels\n",
+    "pip install -q git+https://github.com/makslevental/mlir-python-extras@$BRANCH"
+   ]
   },
   {
    "cell_type": "markdown",

examples/mlir_python_extras.ipynb

@@ -21,10 +21,19 @@
     "# !pip install -q mlir-python-extras -f https://makslevental.github.io/wheels\n",
     "import os\n",
     "BRANCH = os.getenv(\"BRANCH\", \"main\")\n",
-    "os.environ[\"SCRIPT_ADDRESS\"] = f\"https://raw.githubusercontent.com/makslevental/mlir-python-extras/refs/heads/{BRANCH}/scripts/get_latest_bindings.py\"\n",
-    "!curl $SCRIPT_ADDRESS -o get_latest_bindings.py\n",
-    "!latest_version=$(python get_latest_bindings.py \"none\") && pip install -q mlir_python_bindings==$latest_version -f https://makslevental.github.io/wheels\n",
-    "!pip install -q git+https://github.com/makslevental/mlir-python-extras@$BRANCH"
+    "os.environ[\"SCRIPT_ADDRESS\"] = f\"https://raw.githubusercontent.com/makslevental/mlir-python-extras/refs/heads/{BRANCH}/scripts/get_latest_bindings.py\""
+   ]
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": [
+    "%%bash\n",
+    "curl $SCRIPT_ADDRESS -o get_latest_bindings.py\n",
+    "latest_version=$(python get_latest_bindings.py \"none\") && pip install -q mlir_python_bindings==$latest_version -f https://makslevental.github.io/wheels\n",
+    "pip install -q git+https://github.com/makslevental/mlir-python-extras@$BRANCH"
    ]
   },
   {

examples/vectorization_e2e.ipynb

@@ -32,12 +32,7 @@
    ]
   },
   {
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2025-05-15T22:45:29.673103Z",
-     "start_time": "2025-05-15T22:45:18.737745Z"
-    }
-   },
+   "metadata": {},
    "cell_type": "code",
    "source": [
     "# if running by yourself, you can use this instead\n",
@@ -49,18 +44,8 @@
     "!latest_version=$(python get_latest_bindings.py \"none\") && pip install -q mlir_python_bindings==$latest_version -f https://makslevental.github.io/wheels\n",
     "!pip install -q git+https://github.com/makslevental/mlir-python-extras@$BRANCH\n"
    ],
-   "outputs": [
-    {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "  % Total    % Received % Xferd  Average Speed   Time    Time     Time  Current\r\n",
-      "                                 Dload  Upload   Total   Spent    Left  Speed\r\n",
-      "100  2421  100  2421    0     0  15636      0 --:--:-- --:--:-- --:--:-- 15720\r\n"
-     ]
-    }
-   ],
-   "execution_count": 1
+   "outputs": [],
+   "execution_count": null
   },
   {
    "metadata": {},
@@ -70,8 +55,6 @@
   {
    "metadata": {},
    "cell_type": "code",
-   "outputs": [],
-   "execution_count": null,
    "source": [
     "import numpy as np\n",
     "\n",
@@ -104,7 +87,9 @@
     "from mlir.extras.dialects.ext import transform\n",
     "from mlir.extras.runtime.passes import Pipeline, run_pipeline\n",
     "from mlir.extras.runtime.refbackend import LLVMJITBackend\n"
-   ]
+   ],
+   "outputs": [],
+   "execution_count": null
   },
   {
    "cell_type": "markdown",