Fix enum validation error msgs

Author: VaghinakDev

src/superannotate/__init__.py

@@ -1,7 +1,7 @@
 import os
 import sys
 
-__version__ = "4.4.4dev1"
+__version__ = "4.4.4dev2"
 
 sys.path.append(os.path.split(os.path.realpath(__file__))[0])
 

src/superannotate/lib/app/interface/types.py

@@ -5,6 +5,7 @@
 from typing import Union
 
 from lib.core.enums import AnnotationStatus
+from lib.core.enums import BaseTitledEnum
 from lib.core.enums import ClassTypeEnum
 from lib.core.enums import ProjectStatus
 from lib.core.enums import ProjectType
@@ -32,7 +33,11 @@ class EnumMemberError(PydanticTypeError):
     code = "enum"
 
     def __str__(self) -> str:
-        permitted = ", ".join(str(v.name) for v in self.enum_values)  # type: ignore
+        enum_values = list(self.enum_values)  # noqa
+        if isinstance(enum_values[0], BaseTitledEnum):
+            permitted = ", ".join(str(v.name) for v in enum_values)  # type: ignore
+        else:
+            permitted = ", ".join(str(v.value) for v in enum_values)  # type: ignore
         return f"Available values are: {permitted}"
 
 

src/superannotate/lib/app/serializers.py

@@ -121,8 +121,13 @@ def serialize(
         exclude: Set[str] = None,
     ):
 
-        to_exclude = {"sync_status": True, "unverified_users": True,
-                   "classes": {"__all__": {"attribute_groups": {"__all__": {"is_multiselect"}}}}}
+        to_exclude = {
+            "sync_status": True,
+            "unverified_users": True,
+            "classes": {
+                "__all__": {"attribute_groups": {"__all__": {"is_multiselect"}}}
+            },
+        }
         if exclude:
             for field in exclude:
                 to_exclude[field] = True

src/superannotate/lib/core/entities/classes.py

@@ -96,7 +96,7 @@ def __hash__(self):
         return hash(f"{self.id}{self.type}{self.name}")
 
     class Config:
-        extra = Extra.ignore
+        extra = Extra.allow
         json_encoders = {
             HexColor: lambda v: v.__root__,
             BaseTitledEnum: lambda v: v.value,