From ab55cff50259b0cb4662c22bc2e70f0e5165cf2b Mon Sep 17 00:00:00 2001 From: Paul Nechifor Date: Fri, 9 Jan 2026 03:21:19 +0200 Subject: [PATCH 1/2] fix mujoco menagerie --- dimos/agents/vlm_agent.py | 2 +- dimos/protocol/mcp/test_mcp_module.py | 1 - dimos/robot/unitree_webrtc/mujoco_connection.py | 5 +++-- dimos/stream/video_operators.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) diff --git a/dimos/agents/vlm_agent.py b/dimos/agents/vlm_agent.py index 0757a59d22..2600a7ab50 100644 --- a/dimos/agents/vlm_agent.py +++ b/dimos/agents/vlm_agent.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from langchain_core.messages import AIMessage, HumanMessage, SystemMessage +from langchain_core.messages import AIMessage, HumanMessage from dimos.agents.llm_init import build_llm, build_system_message from dimos.agents.spec import AgentSpec, AnyMessage diff --git a/dimos/protocol/mcp/test_mcp_module.py b/dimos/protocol/mcp/test_mcp_module.py index 1deb5b9057..68d6a137f8 100644 --- a/dimos/protocol/mcp/test_mcp_module.py +++ b/dimos/protocol/mcp/test_mcp_module.py @@ -21,7 +21,6 @@ import socket import subprocess import sys -import threading import pytest diff --git a/dimos/robot/unitree_webrtc/mujoco_connection.py b/dimos/robot/unitree_webrtc/mujoco_connection.py index 586f4d0ea7..d88673b9b3 100644 --- a/dimos/robot/unitree_webrtc/mujoco_connection.py +++ b/dimos/robot/unitree_webrtc/mujoco_connection.py @@ -61,9 +61,10 @@ def __init__(self, global_config: GlobalConfig) -> None: # Pre-download the mujoco_sim data. get_data("mujoco_sim") - # Trigger the download of the mujoco_menajerie package. This is so it + # Trigger the download of the mujoco_menagerie package. This is so it # doesn't trigger in the mujoco process where it can time out. - import mujoco_playground + from mujoco_playground._src import mjx_env + mjx_env.ensure_menagerie_exists() self.global_config = global_config self.process: subprocess.Popen[bytes] | None = None diff --git a/dimos/stream/video_operators.py b/dimos/stream/video_operators.py index 558972e155..548bba7598 100644 --- a/dimos/stream/video_operators.py +++ b/dimos/stream/video_operators.py @@ -16,7 +16,7 @@ from collections.abc import Callable from datetime import datetime, timedelta from enum import Enum -from typing import TYPE_CHECKING, Any +from typing import TYPE_CHECKING import cv2 import numpy as np From 7c87858555eb6aae562ee6375680374af3ea634d Mon Sep 17 00:00:00 2001 From: paul-nechifor <1262969+paul-nechifor@users.noreply.github.com> Date: Fri, 9 Jan 2026 02:31:08 +0000 Subject: [PATCH 2/2] CI code cleanup --- dimos/robot/unitree_webrtc/mujoco_connection.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dimos/robot/unitree_webrtc/mujoco_connection.py b/dimos/robot/unitree_webrtc/mujoco_connection.py index d88673b9b3..21a5cf07ec 100644 --- a/dimos/robot/unitree_webrtc/mujoco_connection.py +++ b/dimos/robot/unitree_webrtc/mujoco_connection.py @@ -64,6 +64,7 @@ def __init__(self, global_config: GlobalConfig) -> None: # Trigger the download of the mujoco_menagerie package. This is so it # doesn't trigger in the mujoco process where it can time out. from mujoco_playground._src import mjx_env + mjx_env.ensure_menagerie_exists() self.global_config = global_config