fix optional without None for pydantic 2.0

airo-ugent · Dec 11, 2023 · a7e40e9 · a7e40e9
1 parent 63b8ef2
commit a7e40e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/airo-dataset-tools/airo_dataset_tools/data_parsers/coco.py b/airo-dataset-tools/airo_dataset_tools/data_parsers/coco.py
@@ -103,7 +103,7 @@ class CocoInstanceAnnotation(BaseModel, extra="allow"):  # allow extra fields, t
     image_id: ImageID
     category_id: CategoryID
 
-    bbox: Optional[Tuple[float, float, float, float]]
+    bbox: Optional[Tuple[float, float, float, float]] = None
 
     segmentation: Optional[Segmentation] = None
     area: Optional[float] = None
@@ -120,7 +120,7 @@ def iscrowd_must_be_binary(cls, v: IsCrowd) -> IsCrowd:
 
 class CocoKeypointAnnotation(CocoInstanceAnnotation):
     keypoints: Keypoints
-    num_keypoints: Optional[int]
+    num_keypoints: Optional[int] = None
 
     # make bbox optional
     bbox: Optional[Tuple[float, float, float, float]] = None