From 11d5c58d0f9dde4b13646814ba3b7061fb48a47c Mon Sep 17 00:00:00 2001 From: Dayanne Fernandes Date: Tue, 6 Aug 2024 20:22:28 -0300 Subject: [PATCH] fix linting --- vision_agent/tools/tool_types.py | 6 +++--- vision_agent/tools/tools.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/vision_agent/tools/tool_types.py b/vision_agent/tools/tool_types.py index c575cec8..8ab55ffe 100644 --- a/vision_agent/tools/tool_types.py +++ b/vision_agent/tools/tool_types.py @@ -1,13 +1,13 @@ -from typing import List, Tuple +from typing import List, Tuple, Literal -from nptyping import UInt8, NDArray, Shape +from nptyping import UInt8, NDArray from pydantic import BaseModel, ConfigDict class BboxInput(BaseModel): model_config = ConfigDict(arbitrary_types_allowed=True) - image: NDArray[Shape["Height, Width, 3"], UInt8] + image: NDArray[Literal["Height, Width, 3"], UInt8] filename: str labels: List[str] bboxes: List[Tuple[int, int, int, int]] diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py index 6e323b79..fc5a6d72 100644 --- a/vision_agent/tools/tools.py +++ b/vision_agent/tools/tools.py @@ -879,6 +879,7 @@ def generate_pose_image(image: np.ndarray) -> np.ndarray: data = { "image": image_b64, "tool": "generate_pose", + "function_name": "generate_pose_image", } answer = send_inference_request(data, "tools")