Change schemas

Author: dshabin

src/superannotate_schemas/schemas/custom_types.py

@@ -1,7 +1,7 @@
 from typing import Any
 from typing import TypeVar
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import ValidationError
 from pydantic.error_wrappers import ErrorWrapper
 

src/superannotate_schemas/schemas/external/document.py

@@ -6,7 +6,7 @@
 from superannotate_schemas.schemas.base import BaseMetadata as Metadata
 from superannotate_schemas.schemas.base import Tag
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import Field
 from pydantic import StrictStr
 from pydantic import StrictInt

src/superannotate_schemas/schemas/external/pixel.py

@@ -10,7 +10,7 @@
 from superannotate_schemas.schemas.base import Tag
 from superannotate_schemas.schemas.base import Comment
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import Field
 
 

src/superannotate_schemas/schemas/external/vector.py

@@ -2,7 +2,7 @@
 from typing import Optional
 from typing import Union
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import Field
 from pydantic import StrictFloat
 from pydantic import StrictInt

src/superannotate_schemas/schemas/external/video.py

@@ -14,7 +14,7 @@
 from superannotate_schemas.schemas.base import Tag
 from superannotate_schemas.schemas.base import INVALID_DICT_MESSAGE
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import StrictBool
 from pydantic import conlist
 from pydantic import ValidationError

src/superannotate_schemas/validators.py

@@ -17,7 +17,7 @@
 from superannotate_schemas.schemas.internal import DocumentAnnotation as InternalDocumentAnnotation
 
 
-from pydantic import BaseModel
+from superannotate_schemas.schemas.base import BaseModel
 from pydantic import Extra
 from pydantic import ValidationError
 

tests/test_validators.py

@@ -325,7 +325,7 @@ def test_validate_document_annotation_wrong_class_id(self, mock_print):
             validator = AnnotationValidators.get_validator("document")(data)
             self.assertFalse(validator.is_valid())
             self.assertEqual(validator.generate_report(),
-                             "instances[0].classId                             value is not a valid integer")
+                             "instances[0].classId                             integer type expected")
 
     def test_validate_document_annotation_with_null_created_at(self):
         with tempfile.TemporaryDirectory() as tmpdir_name:
@@ -1297,7 +1297,7 @@ def test_validate_vector_template_polygon_polyline_min_annotation(self, mock_pri
                 data = json.loads(f.read())
             validator = AnnotationValidators.get_validator("vector")(data)
             self.assertFalse(validator.is_valid())
-            self.assertEqual(len(validator.generate_report()),253)
+            self.assertEqual(len(validator.generate_report()),246)
 
 
     def test_validate_video_point_labels(self):
@@ -1972,7 +1972,7 @@ def test_validate_video_point_labels_bad_keys(self):
                 data = json.loads(f.read())
             validator = AnnotationValidators.get_validator("video")(data)
             self.assertFalse(validator.is_valid())
-            self.assertEqual(len(validator.generate_report()),416)
+            self.assertEqual(len(validator.generate_report()),409)
 
     def test_validate_vector_empty_annotation(self):
         with tempfile.TemporaryDirectory() as tmpdir_name:
@@ -2132,7 +2132,7 @@ def test_validate_document_wrong_meta_data(self):
             validator.is_valid()
             print(validator.generate_report())
             self.assertFalse(validator.is_valid())
-            self.assertEqual(len(validator.generate_report()), 155)
+            self.assertEqual(len(validator.generate_report()), 141)
 
     def test_validate_vector_empty_annotation_bad_role(self):
         with tempfile.TemporaryDirectory() as tmpdir_name: