Complete fix

Author: Leo-Expose

google_colab.ipynb

@@ -71,22 +71,46 @@
    "outputs": [],
    "source": [
     "# \u2500\u2500 Install dependencies (Google Colab) \u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\n",
-    "# unsloth[colab-new] is designed for Colab; installs should be clean here.\n",
-    "# If you see resolver warnings (e.g. fsspec/s3fs), they are safe to ignore.\n",
-    "# --no-deps was skipping unsloth_zoo and breaking `import unsloth` / `import train`\n",
-    "!pip install \"unsloth[colab-new]@git+https://github.com/unslothai/unsloth.git\" --quiet\n",
-    "!pip install -U unsloth_zoo --quiet\n",
-    "!pip install trl openenv transformers datasets accelerate matplotlib --quiet\n",
-    "!pip install --upgrade bitsandbytes --quiet\n",
-    "\n",
-    "# Colab-specific: ensure we have the latest huggingface_hub\n",
-    "!pip install --upgrade huggingface_hub --quiet\n",
-    "\n",
-    "# Suppress tokenizer parallelism warning\n",
+    "# Use sys.executable so packages land in *this* kernel (plain !pip can target another Python).\n",
+    "# Order: generic stack first, then Unsloth, then unsloth_zoo last (pip resolver won\u2019t strip it).\n",
+    "\n",
+    "import subprocess\n",
+    "import sys\n",
+    "\n",
+    "\n",
+    "def pip(*args: str) -> None:\n",
+    "    cmd = [sys.executable, \"-m\", \"pip\", *args]\n",
+    "    print(\" \", \" \".join(cmd))\n",
+    "    subprocess.check_call(cmd)\n",
+    "\n",
+    "\n",
+    "pip(\"install\", \"-q\", \"trl\", \"openenv\", \"transformers\", \"datasets\", \"accelerate\", \"matplotlib\")\n",
+    "pip(\"install\", \"-q\", \"--upgrade\", \"bitsandbytes\", \"huggingface_hub\")\n",
+    "\n",
+    "pip(\n",
+    "    \"install\",\n",
+    "    \"-q\",\n",
+    "    \"unsloth[colab-new]@git+https://github.com/unslothai/unsloth.git\",\n",
+    ")\n",
+    "pip(\"install\", \"-q\", \"-U\", \"unsloth_zoo>=2026.4.8\")\n",
+    "\n",
+    "import importlib.metadata as _im\n",
+    "\n",
+    "try:\n",
+    "    _z = _im.version(\"unsloth_zoo\")\n",
+    "except _im.PackageNotFoundError:\n",
+    "    print(\"\u26a0\ufe0f  PyPI metadata missing; installing unsloth_zoo from GitHub\u2026\")\n",
+    "    pip(\"install\", \"-q\", \"-U\", \"git+https://github.com/unslothai/unsloth-zoo.git\")\n",
+    "    _z = _im.version(\"unsloth_zoo\")\n",
+    "\n",
+    "print(f\"\u2705 unsloth_zoo {_z} (importlib metadata OK)\")\n",
+    "\n",
     "import os\n",
-    "os.environ['TOKENIZERS_PARALLELISM'] = 'false'\n",
     "\n",
-    "print('\u2705 Dependencies installed')\n"
+    "os.environ[\"TOKENIZERS_PARALLELISM\"] = \"false\"\n",
+    "\n",
+    "print(\"\u2705 Dependencies installed.\")\n",
+    "print(\"   If a later cell still errors on unsloth, use: Runtime \u2192 Restart session, then re-run this cell.\")\n"
    ]
   },
   {