Add polygon polyline to video

Author: VaghinakDev

src/superannotate_schemas/__init__.py

@@ -6,7 +6,7 @@
 
 from superannotate_schemas.validators import AnnotationValidators
 
-__version__ = '1.0.43dev5'
+__version__ = '1.0.44dev1'
 
 __all__ = [
     "__version__",

src/superannotate_schemas/schemas/external/vector.py

@@ -9,7 +9,6 @@
 from pydantic import conlist
 from pydantic.error_wrappers import ErrorWrapper
 
-from superannotate_schemas.schemas.base import StrictPointNumber
 from superannotate_schemas.schemas.base import AxisPoint
 from superannotate_schemas.schemas.base import BaseAttribute
 from superannotate_schemas.schemas.base import BaseImageMetadata
@@ -21,6 +20,7 @@
 from superannotate_schemas.schemas.base import INVALID_DICT_MESSAGE
 from superannotate_schemas.schemas.base import NotEmptyStr
 from superannotate_schemas.schemas.base import StrictNumber
+from superannotate_schemas.schemas.base import StrictPointNumber
 from superannotate_schemas.schemas.base import Tag
 from superannotate_schemas.schemas.enums import VectorAnnotationTypeEnum
 

src/superannotate_schemas/schemas/external/video.py

@@ -143,7 +143,7 @@ class PointInstance(BaseModel):
 
 class PolygonInstance(BaseModel):
     meta: PolygonInstanceMetadata
-    parameters: conlist(PolylineParameter, min_items=1)
+    parameters: conlist(PolygonParameter, min_items=1)
 
 
 class PolylineInstance(BaseModel):
@@ -159,7 +159,9 @@ class EventInstance(BaseModel):
 ANNOTATION_TYPES = {
     VideoType.BBOX: BboxInstance,
     VideoType.EVENT: EventInstance,
-    VideoType.POINT: PointInstance
+    VideoType.POINT: PointInstance,
+    VideoType.POLYGON: PolygonInstance,
+    VideoType.POLYLINE: PolylineInstance
 }