diff --git a/tests/tools/test_tool_utils.py b/tests/tools/test_tool_utils.py index 6a5d8e3ac..7265df5e7 100644 --- a/tests/tools/test_tool_utils.py +++ b/tests/tools/test_tool_utils.py @@ -16,7 +16,7 @@ from typing import Any, Dict, List, Type import pytest -from geometry_msgs.msg import Point +from geometry_msgs.msg import Point, TransformStamped from langchain_core.messages import AIMessage, ToolCall from langchain_core.tools import BaseTool from nav2_msgs.action import NavigateToPose @@ -72,6 +72,7 @@ def _run(self, tool_input: TestToolInput) -> str: Point(), Image(), TFMessage(), + TransformStamped(), NavigateToPose.Goal(), NavigateToPose.Result(), NavigateToPose.Feedback(),