dev8

Author: VaghinakDev

src/superannotate/__init__.py

@@ -1,7 +1,7 @@
 import os
 import sys
 
-__version__ = "4.4.2dev7"
+__version__ = "4.4.2dev8"
 
 sys.path.append(os.path.split(os.path.realpath(__file__))[0])
 

src/superannotate/lib/app/analytics/aggregators.py

@@ -1,13 +1,13 @@
 import copy
 import json
+from dataclasses import dataclass
 from pathlib import Path
 from typing import List
 from typing import Optional
 from typing import Union
 
 import lib.core as constances
 import pandas as pd
-from dataclasses import dataclass
 from lib.app.exceptions import AppException
 from lib.core import ATTACHED_VIDEO_ANNOTATION_POSTFIX
 from lib.core import PIXEL_ANNOTATION_POSTFIX
@@ -99,7 +99,7 @@ class DataAggregator:
             ry=annotation["ry"],
             angle=annotation["angle"],
         ),
-        "tag": lambda annotation: None
+        "tag": lambda annotation: None,
     }
 
     def __init__(

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

@@ -579,8 +579,12 @@ def search_annotation_classes(
         :rtype: list of dicts
         """
         project_name, folder_name = extract_project_folder(project)
-        classes = self.controller.search_annotation_classes(project_name, name_contains)
-        return [i.dict(fill_enum_values=True) for i in classes.data]
+        response = self.controller.search_annotation_classes(
+            project_name, name_contains
+        )
+        if response.errors:
+            raise AppException(response.errors)
+        return [BaseSerializer(i).serialize(exclude_unset=True) for i in response.data]
 
     def set_project_default_image_quality_in_editor(
         self,
@@ -1231,7 +1235,7 @@ def create_annotation_class(
         )
         if response.errors:
             raise AppException(response.errors)
-        return BaseSerializer(response.data).serialize()
+        return BaseSerializer(response.data).serialize(exclude_unset=True)
 
     def delete_annotation_class(
         self, project: NotEmptyStr, annotation_class: Union[dict, NotEmptyStr]
@@ -1308,7 +1312,7 @@ def create_annotation_classes_from_classes_json(
         )
         if response.errors:
             raise AppException(response.errors)
-        return [BaseSerializer(i).serialize() for i in response.data]
+        return [BaseSerializer(i).serialize(exclude_unset=True) for i in response.data]
 
     def download_export(
         self,

src/superannotate/lib/app/serializers.py

@@ -27,9 +27,17 @@ def serialize(
         by_alias: bool = True,
         flat: bool = False,
         exclude: Set[str] = None,
+        exclude_unset=False,
     ):
         return self._fill_enum_values(
-            self._serialize(self._entity, fields, by_alias, flat, exclude=exclude)
+            self._serialize(
+                self._entity,
+                fields,
+                by_alias,
+                flat,
+                exclude=exclude,
+                exclude_unset=exclude_unset,
+            )
         )
 
     def serialize_item(

src/superannotate/lib/core/config.py

@@ -1,8 +1,7 @@
 import threading
-from typing import Dict
-
 from dataclasses import dataclass
 from dataclasses import field
+from typing import Dict
 
 
 class Session:

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

@@ -98,4 +98,5 @@ def __hash__(self):
     class Config:
         validate_assignment = True
         exclude_none = True
+        exclude_unset = True
         fill_enum_values = True