Prajwal782007 commited on
Commit
c220c03
·
1 Parent(s): 84396d2

fix: order imports in Step 1 and add missing torch import in Step 7

Browse files
Files changed (1) hide show
  1. scripts/gridmind_grpo_colab.ipynb +2 -1
scripts/gridmind_grpo_colab.ipynb CHANGED
@@ -60,8 +60,8 @@
60
  "source": [
61
  "import requests\n",
62
  "import json\n",
63
- "import time\n",
64
  "import sys\n",
 
65
  "\n",
66
  "ENV_URL = \"https://prajwal782007-gridmind.hf.space\"\n",
67
  "\n",
@@ -493,6 +493,7 @@
493
  "metadata": {},
494
  "outputs": [],
495
  "source": [
 
496
  "def run_llm_episode(task_id=1, max_steps=96):\n",
497
  " \"\"\"Run an episode using the trained LLM.\"\"\"\n",
498
  " try:\n",
 
60
  "source": [
61
  "import requests\n",
62
  "import json\n",
 
63
  "import sys\n",
64
+ "import time\n",
65
  "\n",
66
  "ENV_URL = \"https://prajwal782007-gridmind.hf.space\"\n",
67
  "\n",
 
493
  "metadata": {},
494
  "outputs": [],
495
  "source": [
496
+ "import torch\n\n",
497
  "def run_llm_episode(task_id=1, max_steps=96):\n",
498
  " \"\"\"Run an episode using the trained LLM.\"\"\"\n",
499
  " try:\n",