Update train_efficient.py: auto-download from HF Hub for job env

train_efficient.py

@@ -24,14 +24,25 @@ from torch.distributions import Normal
 # ============================================================
 # Import the base agent as a module-level namespace
 # ============================================================
+# Download submission.py from HF Hub if not available locally
+import urllib.request
+_SUBMISSION_PATH = '/app/submission.py'
+if not os.path.exists(_SUBMISSION_PATH):
+    print("Downloading submission.py from HF Hub...")
+    urllib.request.urlretrieve(
+        "https://huggingface.co/Builder-Neekhil/orbit-wars-agent/resolve/main/submission.py",
+        _SUBMISSION_PATH
+    )
+    print("Downloaded.")
+
 sys.path.insert(0, '/app')
 _BASE_NS = {}
-exec(open('/app/submission.py').read(), _BASE_NS)
+exec(open(_SUBMISSION_PATH).read(), _BASE_NS)
 print("Base agent loaded successfully.")
 
 # Also create a separate namespace for the opponent
 _OPP_NS = {}
-exec(open('/app/submission.py').read(), _OPP_NS)
+exec(open(_SUBMISSION_PATH).read(), _OPP_NS)
 
 from kaggle_environments import make as _make_env
 
@@ -608,10 +619,17 @@ def train():
         if not best_path.exists():
             best_path = final_path
         if best_path.exists():
+            # Download generate_submission.py from HF Hub
+            gen_script = '/app/generate_submission.py'
+            if not os.path.exists(gen_script):
+                urllib.request.urlretrieve(
+                    "https://huggingface.co/Builder-Neekhil/orbit-wars-agent/resolve/main/generate_submission.py",
+                    gen_script
+                )
             sys.path.insert(0, '/app')
             from generate_submission import generate_submission
             generate_submission(
-                base_agent_path="/app/submission.py",
+                base_agent_path=_SUBMISSION_PATH,
                 checkpoint_path=str(best_path),
                 output_path="/app/submission_adaptive.py",
             )