tod

Author: VaghinakDev

src/superannotate_schemas/schemas/base.py

@@ -26,7 +26,7 @@
 from superannotate_schemas.schemas.enums import BaseImageRoleEnum
 from superannotate_schemas.schemas.enums import VectorAnnotationTypeEnum
 from superannotate_schemas.schemas.enums import AnnotationStatusEnum
-from superannotate_schemas.schemas.enums import TagTypeEnum
+from superannotate_schemas.schemas.enums import ClassTypeEnum
 from superannotate_schemas.schemas.constances import DATE_REGEX
 from superannotate_schemas.schemas.constances import DATE_TIME_FORMAT_ERROR_MESSAGE
 from superannotate_schemas.schemas.constances import POINT_LABEL_VALUE_FORMAT_ERROR_MESSAGE
@@ -122,7 +122,7 @@ class BaseInstance(TrackableModel, TimedBaseModel):
 
 
 class BaseInstanceTag(BaseInstance):
-    type: TagTypeEnum
+    type: ClassTypeEnum
     probability: Optional[StrictInt] = Field(100)
     attributes: Optional[List[BaseAttribute]] = Field(list())
 

src/superannotate_schemas/schemas/classes.py

@@ -2,7 +2,6 @@
 from typing import List
 
 from superannotate_schemas.schemas.base import BaseModel
-from superannotate_schemas.schemas.base import TrackableModel
 from superannotate_schemas.schemas.base import TimedBaseModel
 from superannotate_schemas.schemas.base import StrictInt
 from superannotate_schemas.schemas.base import StrictStr

src/superannotate_schemas/validators.py

@@ -15,6 +15,7 @@
 from superannotate_schemas.schemas.internal import VectorAnnotation as InternalVectorAnnotation
 from superannotate_schemas.schemas.internal import VideoAnnotation as InternalVideoAnnotation
 from superannotate_schemas.schemas.internal import DocumentAnnotation as InternalDocumentAnnotation
+from superannotate_schemas.schemas.classes import AnnotationClass
 
 
 from superannotate_schemas.schemas.base import BaseModel
@@ -95,7 +96,6 @@ class AnnotationValidators:
             ValidatorFactory.get_validator(ExternalDocumentAnnotation),
             ValidatorFactory.get_validator(InternalDocumentAnnotation)
         )
-
     }
 
     @classmethod