Upload LogTriageEnv_Training.ipynb

LogTriageEnv_Training.ipynb

@@ -180,12 +180,47 @@
    "source": [
     "import subprocess\n",
     "import os\n",
+    "import shutil\n",
     "\n",
-    "# Ensure train.py exists in current directory\n",
-    "if not os.path.exists('train.py'):\n",
-    "    print(\"⚠ train.py not found in current directory\")\n",
-    "    print(\"Make sure to upload train.py to this notebook first\")\n",
-    "else:\n",
+    "print(\"\\n\" + \"=\"*60)\n",
+    "print(\"[STEP 5A] Clone Repository from GitHub\")\n",
+    "print(\"=\"*60)\n",
+    "\n",
+    "# Clone the repository\n",
+    "repo_url = \"https://github.com/rohitdecodes/logtriage-env.git\"\n",
+    "repo_dir = \"logtriage-env\"\n",
+    "\n",
+    "# Remove existing repo if it exists\n",
+    "if os.path.exists(repo_dir):\n",
+    "    print(f\"⚠ {repo_dir} already exists, removing...\")\n",
+    "    shutil.rmtree(repo_dir)\n",
+    "\n",
+    "try:\n",
+    "    print(f\"Cloning from {repo_url}...\")\n",
+    "    result = subprocess.run(\n",
+    "        [\"git\", \"clone\", repo_url, repo_dir],\n",
+    "        capture_output=True,\n",
+    "        text=True,\n",
+    "        timeout=300\n",
+    "    )\n",
+    "\n",
+    "    if result.returncode == 0:\n",
+    "        print(f\"✓ Repository cloned successfully\")\n",
+    "        train_py_path = os.path.join(repo_dir, \"train.py\")\n",
+    "    else:\n",
+    "        print(f\"⚠ Clone failed: {result.stderr}\")\n",
+    "        train_py_path = \"train.py\"\n",
+    "except Exception as e:\n",
+    "    print(f\"⚠ Clone error: {e}\")\n",
+    "    train_py_path = \"train.py\"\n",
+    "\n",
+    "print()\n",
+    "print(\"=\"*60)\n",
+    "print(\"[STEP 5B] Launch Training\")\n",
+    "print(\"=\"*60)\n",
+    "\n",
+    "# Check if train.py exists (either from clone or current directory)\n",
+    "if os.path.exists(train_py_path):\n",
     "    print(\"\\n\" + \"=\"*60)\n",
     "    print(\"[START] LogTriageEnv Training\")\n",
     "    print(\"=\"*60)\n",
@@ -196,7 +231,7 @@
     "    print()\n",
     "\n",
     "    cmd = [\n",
-    "        \"python\", \"train.py\",\n",
+    "        \"python\", train_py_path,\n",
     "        \"--model\", model_id,\n",
     "        \"--task\", \"all\",\n",
     "        \"--episodes\", \"50\",\n",
@@ -216,7 +251,11 @@
     "    except subprocess.TimeoutExpired:\n",
     "        print(\"⚠ Training timed out after 30 minutes\")\n",
     "    except Exception as e:\n",
-    "        print(f\"Error: {e}\")"
+    "        print(f\"Error: {e}\")\n",
+    "else:\n",
+    "    print(f\"⚠ train.py not found at {train_py_path}\")\n",
+    "    print(\"✗ TRAINING FAILED\")\n",
+    "    print(\"Make sure the repository clone was successful or train.py exists in current directory\")"
    ]
   },
   {
@@ -343,7 +382,8 @@
     "        df = pd.read_csv(csv_file)\n",
     "        print(f\"[{files[0]}]\")\n",
     "        print(df.head(10).to_string())\n",
-    "        print(f\"\\n✓ {len(df)} episodes tracked\")\nelse:\n",
+    "        print(f\"\\n✓ {len(df)} episodes tracked\")\n",
+    "else:\n",
     "    print(f\"⚠ Log directory not found: {csv_dir}\")\n",
     "    print(\"CSV logs are generated during training\")"
    ]
@@ -384,7 +424,8 @@
     "                !zip -r {f}.zip {f}\n",
     "                files.download(f\"{f}.zip\")\n",
     "            print(f\"✓ {f} ready\")\n",
-    "        \nexcept ImportError:\n",
+    "        \n",
+    "except ImportError:\n",
     "    print(\"[INFO] Not in Colab environment\")\n",
     "    print(\"Files saved locally:\")\n",
     "    !ls -lh reward_curve.png logtriage-trained/ phase2_checkpoints/ logs/ 2>/dev/null || echo \"Check current directory\""