From 48f0eb8dd013181e98a350a7b77255b21b7b9ab2 Mon Sep 17 00:00:00 2001 From: Logic Date: Thu, 2 Apr 2026 08:26:39 +0800 Subject: [PATCH] fix: defer MuJoCo env import until headless GL is configured --- roboimi/demos/vla_scripts/eval_vla.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/roboimi/demos/vla_scripts/eval_vla.py b/roboimi/demos/vla_scripts/eval_vla.py index 2ada76c..a237d4f 100644 --- a/roboimi/demos/vla_scripts/eval_vla.py +++ b/roboimi/demos/vla_scripts/eval_vla.py @@ -25,7 +25,6 @@ from omegaconf import DictConfig, OmegaConf from hydra.utils import instantiate 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.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') +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( ckpt_path: str, agent_cfg: DictConfig,