training/train_grpo.ipynb: add Kaggle (/kaggle/working) fresh-clone branch

training/train_grpo.ipynb

@@ -42,7 +42,7 @@
       "cell_type": "code",
       "metadata": {},
       "source": [
-        "# Cell 2: Resolve repo path (Colab: fresh clone. Local: auto-detect project root)\n",
+        "# Cell 2: Resolve repo path (Colab / Kaggle: fresh clone. Local: auto-detect project root)\n",
         "import os\n",
         "import sys\n",
         "import shutil\n",
@@ -52,6 +52,7 @@
         "REPO_BRANCH = \"main\"\n",
         "REPO_URL = \"https://github.com/VaibhavKhandare/viral-posts-env.git\"\n",
         "COLAB_REPO = Path(\"/content/viral-posts-env\")\n",
+        "KAGGLE_REPO = Path(\"/kaggle/working/viral-posts-env\")\n",
         "\n",
         "\n",
         "def _is_repo_root(p: Path) -> bool:\n",
@@ -64,33 +65,39 @@
         "        if _is_repo_root(cand):\n",
         "            return cand\n",
         "    raise FileNotFoundError(\n",
-        "        \"Could not find project root. cd into viral-posts-env or run this notebook in Google Colab.\"\n",
+        "        \"Could not find project root. cd into viral-posts-env or run this notebook in Google Colab/Kaggle.\"\n",
         "    )\n",
         "\n",
         "\n",
-        "# --- Colab: always clone a clean copy (avoids stale 7-day code) ---\n",
-        "if Path(\"/content\").is_dir():\n",
-        "    if COLAB_REPO.exists():\n",
-        "        shutil.rmtree(COLAB_REPO, ignore_errors=True)\n",
+        "def _fresh_clone(target: Path) -> None:\n",
+        "    if target.exists():\n",
+        "        shutil.rmtree(target, ignore_errors=True)\n",
+        "    target.parent.mkdir(parents=True, exist_ok=True)\n",
         "    p = subprocess.run(\n",
-        "        [\n",
-        "            \"git\", \"clone\", \"--branch\", REPO_BRANCH, \"--depth\", \"1\",\n",
-        "            REPO_URL, str(COLAB_REPO),\n",
-        "        ],\n",
-        "        capture_output=True,\n",
-        "        text=True,\n",
+        "        [\"git\", \"clone\", \"--branch\", REPO_BRANCH, \"--depth\", \"1\", REPO_URL, str(target)],\n",
+        "        capture_output=True, text=True,\n",
         "    )\n",
         "    if p.returncode != 0:\n",
         "        raise RuntimeError(\n",
-        "            \"git clone failed. Check network and branch name.\\n\"\n",
+        "            \"git clone failed. On Kaggle, enable Internet in the notebook settings panel.\\n\"\n",
         "            f\"stdout:\\n{p.stdout}\\nstderr:\\n{p.stderr}\"\n",
         "        )\n",
-        "    if not COLAB_REPO.is_dir():\n",
-        "        raise FileNotFoundError(f\"Clone did not create {COLAB_REPO}\")\n",
+        "    if not target.is_dir():\n",
+        "        raise FileNotFoundError(f\"Clone did not create {target}\")\n",
+        "\n",
+        "\n",
+        "_IS_KAGGLE = bool(os.environ.get(\"KAGGLE_KERNEL_RUN_TYPE\")) or Path(\"/kaggle/working\").is_dir()\n",
+        "_IS_COLAB = (not _IS_KAGGLE) and Path(\"/content\").is_dir()\n",
+        "\n",
+        "if _IS_KAGGLE:\n",
+        "    _fresh_clone(KAGGLE_REPO)\n",
+        "    os.chdir(KAGGLE_REPO)\n",
+        "    print(\"Mode: Kaggle (fresh clone)\")\n",
+        "elif _IS_COLAB:\n",
+        "    _fresh_clone(COLAB_REPO)\n",
         "    os.chdir(COLAB_REPO)\n",
         "    print(\"Mode: Colab (fresh clone)\")\n",
         "else:\n",
-        "    # --- Local machine: do not use /content ---\n",
         "    root = _find_local_root()\n",
         "    os.chdir(root)\n",
         "    print(\"Mode: local\")\n",
@@ -1255,7 +1262,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.13.1"
+      "version": "3.14.2"
     }
   },
   "nbformat": 4,