Added NaN handler

Author: VaghinakDev

src/superannotate_schemas/__init__.py

@@ -6,7 +6,7 @@
 
 from superannotate_schemas.validators import AnnotationValidators
 
-__version__ = '1.0.40'
+__version__ = '1.0.42dev1'
 
 __all__ = [
     "__version__",

src/superannotate_schemas/schemas/base.py

@@ -1,5 +1,7 @@
+import datetime
 from typing import Dict
 from typing import List
+from math import isnan
 from typing import Optional
 from typing import Union
 
@@ -8,7 +10,7 @@
 from pydantic import constr
 from pydantic import Extra
 from pydantic import StrictInt
-from pydantic import StrictFloat
+from pydantic import StrictFloat as PydanticStrictFloat
 from pydantic import StrictStr
 from pydantic import StrictBool
 from pydantic import Field
@@ -39,6 +41,20 @@ def enum_error_handling(self) -> str:
     return f"Invalid value, permitted: {permitted}"
 
 
+class StrictFloat(PydanticStrictFloat):
+
+    @classmethod
+    def __get_validators__(cls):
+        yield from super().__get_validators__()
+        yield cls._validate_nan
+
+    @staticmethod
+    def _validate_nan(v):
+        if isnan(v):
+            raise TypeError("NaN is not a valid float")
+        return v
+
+
 EnumMemberError.__str__ = enum_error_handling
 
 NotEmptyStr = constr(strict=True, min_length=1)
@@ -59,6 +75,7 @@ def validate(cls, value: Union[str]) -> Union[str]:
 
 
 class BaseModel(PyDanticBaseModel):
+
     class Config:
         extra = Extra.allow
         use_enum_values = True
@@ -96,14 +113,17 @@ class TimedBaseModel(BaseModel):
     created_at: Optional[constr(regex=DATE_REGEX)] = Field(None, alias="createdAt")
     updated_at: Optional[constr(regex=DATE_REGEX)] = Field(None, alias="updatedAt")
 
-    @validator("created_at", "updated_at", pre=True)
+    @validator("created_at", "updated_at", pre=True, always=True)
     def validate_created_at(cls, value):
-        try:
-            if value is not None:
-                constr(regex=DATE_REGEX, strict=True).validate(value)
-        except (TypeError, StrRegexError):
-            raise TypeError(DATE_TIME_FORMAT_ERROR_MESSAGE)
-        return value
+        if value:
+            try:
+                if value is not None:
+                    constr(regex=DATE_REGEX, strict=True).validate(value)
+            except (TypeError, StrRegexError):
+                raise TypeError(DATE_TIME_FORMAT_ERROR_MESSAGE)
+            return value
+        else:
+            return datetime.datetime.now().strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z"
 
 
 class UserAction(BaseModel):

src/superannotate_schemas/schemas/external/pixel.py

@@ -5,7 +5,6 @@
 from superannotate_schemas.schemas.base import BaseAttribute
 from superannotate_schemas.schemas.base import BaseImageMetadata
 from superannotate_schemas.schemas.base import NotEmptyStr
-from superannotate_schemas.schemas.base import StrictStr
 from superannotate_schemas.schemas.base import HexColor
 from superannotate_schemas.schemas.base import Tag
 from superannotate_schemas.schemas.base import Comment

src/superannotate_schemas/schemas/external/vector.py

@@ -3,13 +3,13 @@
 from typing import Union
 
 from pydantic import Field
-from pydantic import StrictFloat
 from pydantic import StrictInt
 from pydantic import StrictStr
 from pydantic import ValidationError
 from pydantic import conlist
 from pydantic.error_wrappers import ErrorWrapper
 
+from superannotate_schemas.schemas.base import StrictFloat
 from superannotate_schemas.schemas.base import AxisPoint
 from superannotate_schemas.schemas.base import BaseAttribute
 from superannotate_schemas.schemas.base import BaseImageMetadata

src/superannotate_schemas/schemas/internal/vector.py

@@ -18,7 +18,7 @@
 from superannotate_schemas.schemas.base import NotEmptyStr
 
 from pydantic import StrictInt
-from pydantic import StrictFloat
+from superannotate_schemas.schemas.base import StrictFloat
 from pydantic import conlist
 from pydantic import Field
 from pydantic import validate_model

src/superannotate_schemas/schemas/internal/video.py

@@ -13,7 +13,7 @@
 from superannotate_schemas.schemas.base import AnnotationStatusEnum
 
 from superannotate_schemas.schemas.base import BaseModel
-from pydantic import StrictFloat
+from superannotate_schemas.schemas.base import StrictFloat
 from pydantic import constr
 from pydantic import Field
 from pydantic import StrictBool