Update env.py

env.py

@@ -2,11 +2,20 @@
 
 from pathlib import Path
 import os
+import sys
 import errno
 import yaml
 import gymnasium as gym
 from gymnasium import spaces
 import numpy as np
+from huggingface_hub import snapshot_download
+
+# Download the full repo and get its path
+_REPO_PATH = Path(snapshot_download(repo_id='lerobot/unitree-g1-mujoco', revision=None, cache_dir=None))
+
+# Add the repo to sys.path so we can import sim module
+if str(_REPO_PATH) not in sys.path:
+    sys.path.insert(0, str(_REPO_PATH))
 
 from sim.simulator_factory import SimulatorFactory, init_channel
 
@@ -82,7 +91,7 @@ def make_env(
         try:
             # --- run_sim.py: main() body ---
 
-            config_path = Path(__file__).parent / "config.yaml"
+            config_path = _REPO_PATH / "config.yaml"
             with open(config_path) as f:
                 config = yaml.safe_load(f)