Update env.py

env.py

@@ -1,6 +1,15 @@
 import logging
 from typing import Any
-from .isaaclab_env_wrapper import IsaacLabEnvWrapper
+
+import os
+import sys
+
+# Ensure the repository root is on sys.path
+_REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
+if _REPO_ROOT not in sys.path:
+    sys.path.insert(0, _REPO_ROOT)
+
+from isaaclab_env_wrapper import IsaacLabEnvWrapper
 
 
 def make_env(n_envs: int = 1, use_async_envs: bool = False, **kwargs: Any):