Update training/colab_notebook.py

training/colab_notebook.py

@@ -27,8 +27,9 @@ os.environ["ENV_URL"]     = "https://junaid0600-sql-db-engineer-agent.hf.space"
 os.environ["MODEL_NAME"]  = "unsloth/Qwen2.5-7B-Instruct"
 os.environ["OUTPUT_DIR"]  = "./sdea-trained"
 os.environ["N_EPISODES"]  = "10"
+os.environ["MAX_STEPS"]   = "300"
 
-print("✅ Environment variables set")
+print("Environment variables set")
 print(f"ENV_URL: {os.environ['ENV_URL']}")
 
 # ── CELL 4: Verify environment is live ────────────────────
@@ -40,17 +41,17 @@ def check_env():
     try:
         r = requests.get(f"{ENV_URL}/health", timeout=10)
         data = r.json()
-        print(f"✅ Environment healthy: {data}")
+        print(f"Environment healthy: {data}")
 
         r2 = requests.get(f"{ENV_URL}/tasks", timeout=10)
         tasks = r2.json()
-        print(f"✅ Tasks available: {tasks['total']}")
+        print(f"Tasks available: {tasks['total']}")
 
         r3 = requests.get(f"{ENV_URL}/progress", timeout=10)
-        print(f"✅ Progress endpoint: {r3.status_code}")
+        print(f"Progress endpoint: {r3.status_code}")
         return True
     except Exception as e:
-        print(f"❌ Environment check failed: {e}")
+        print(f"Environment check failed: {e}")
         return False
 
 check_env()
@@ -58,7 +59,7 @@ check_env()
 # ── CELL 5: Quick episode test ────────────────────────────
 def test_episode():
     """Run one full episode to verify everything works."""
-    print("\n🧪 Testing full episode...")
+    print("\n Testing full episode...")
 
     # Reset
     r = requests.post(f"{ENV_URL}/reset",
@@ -97,7 +98,7 @@ def test_episode():
         print(f"  Final score: {summary.get('final_score', 'N/A')}")
         print(f"  Improvement: {summary.get('improvement', 'N/A')}")
 
-    print("\n✅ Episode test complete!")
+    print("\n Episode test complete!")
 
 test_episode()
 
@@ -212,4 +213,4 @@ def run_judge_demo():
     print("Trained to think like a senior DBA.")
     print("=" * 60)
 
-run_judge_demo()
+run_judge_demo()