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..21a5cf07ec 100644 --- a/dimos/robot/unitree_webrtc/mujoco_connection.py +++ b/dimos/robot/unitree_webrtc/mujoco_connection.py @@ -61,9 +61,11 @@ 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