fix linter

landing-ai · Aug 13, 2024 · 4bee8d7 · 4bee8d7
1 parent fd87fa5
commit 4bee8d7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -78,6 +78,8 @@ line_length = 88
 profile = "black"
 
 [tool.mypy]
+plugins = "pydantic.mypy"
+
 exclude = "tests"
 show_error_context = true
 pretty = true

diff --git a/vision_agent/tools/meta_tools_types.py b/vision_agent/tools/meta_tools_types.py
@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import List, Tuple, Optional
 
-from pydantic import BaseModel, ConfigDict, Field, field_serializer
+from pydantic import BaseModel, ConfigDict, Field, field_serializer, SerializationInfo
 
 
 class BboxInput(BaseModel):
@@ -37,7 +37,7 @@ class FineTuning(BaseModel):
  job_id: UUID = Field(alias="jobId")
 
  @field_serializer("job_id")
- def serialize_job_id(self, job_id: UUID, _info):
+ def serialize_job_id(self, job_id: UUID, _info: SerializationInfo) -> str:
  return str(job_id)