JiajunLI/roboimi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: defer MuJoCo env import until headless GL is configured

Browse Source
This commit is contained in:
Logic
2026-04-02 08:26:39 +08:00
parent 3a17744dcf
commit 48f0eb8dd0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
roboimi/demos/vla_scripts/eval_vla.py
View File

@@ -25,7 +25,6 @@ from omegaconf import DictConfig, OmegaConf
from hydra.utils import instantiate from hydra.utils import instantiate
from einops import rearrange from einops import rearrange
from roboimi.envs.double_pos_ctrl_env import make_sim_env
from roboimi.utils.act_ex_utils import sample_transfer_pose from roboimi.utils.act_ex_utils import sample_transfer_pose
from roboimi.vla.eval_utils import execute_policy_action from roboimi.vla.eval_utils import execute_policy_action
@@ -48,6 +47,11 @@ def _configure_headless_mujoco_gl(eval_cfg: DictConfig) -> None:
log.info('headless eval detected without DISPLAY; set MUJOCO_GL=egl') log.info('headless eval detected without DISPLAY; set MUJOCO_GL=egl')
def make_sim_env(task_name: str, headless: bool = False):
from roboimi.envs.double_pos_ctrl_env import make_sim_env as _make_sim_env_impl
return _make_sim_env_impl(task_name, headless=headless)
def load_checkpoint( def load_checkpoint(
ckpt_path: str, ckpt_path: str,
agent_cfg: DictConfig, agent_cfg: DictConfig,