fix linting

landing-ai · Aug 6, 2024 · 11d5c58 · 11d5c58
1 parent 57845ee
commit 11d5c58
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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")