Adjust tests.

Author: VaghinakDev

docs/source/superannotate.sdk.rst

@@ -61,6 +61,8 @@ Exports
 _______
 
 .. autofunction:: superannotate.prepare_export
+.. autofunction:: superannotate.get_annotations
+.. autofunction:: superannotate.get_annotations_per_frame
 .. _ref_download_export:
 .. autofunction:: superannotate.download_export
 .. autofunction:: superannotate.get_exports

requirements.txt

@@ -18,3 +18,5 @@ mixpanel==4.8.3
 pydantic>=1.8.2
 pydantic[email]
 setuptools~=57.4.0
+aiohttp==3.8.1
+

requirements_dev.txt

@@ -1,3 +1,2 @@
--r requirements.txt
 superannotate_schemas>=1.0.38.b1
 

requirements_prod.txt

@@ -1,2 +1 @@
--r requirements.txt
 superannotate_schemas=1.0.39

setup.py

@@ -7,21 +7,18 @@
 with open('src/superannotate/version.py') as f:
     version = f.read().rstrip()[15:-1]
 
-requirements_path = f"requirements_{'dev' if parse(version).is_prerelease else 'prod'}.txt"
+requirements_path = "requirements_{}.txt".format('dev' if parse(version).is_prerelease else 'prod')
+requirements = []
 
 with open(requirements_path) as f:
-    requirements = f.read()
-requirements = requirements.splitlines()
+    requirements.extend(f.read().splitlines())
 
-if sys.platform == 'linux':
-    with open('requirements_extra.txt') as f:
-        requirements_extra = f.read()
-
-    requirements_extra = requirements_extra.splitlines()
-    requirements += requirements_extra
+with open("requirements.txt") as f:
+    requirements.extend(f.read().splitlines())
 
 with open('README.md') as f:
     readme = f.read()
+
 readme = "\n".join(readme.split('\n')[2:])
 
 

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

@@ -72,7 +72,6 @@ def create_project(self, name: str, description: str, type: str):
         To create a new project
         """
         create_project(name, description, type)
-        sys.exit(0)
 
     def create_folder(self, project: str, name: str):
         """

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

@@ -44,6 +44,7 @@
 from pydantic import conlist
 from pydantic import parse_obj_as
 from pydantic import StrictBool
+from pydantic.error_wrappers import ValidationError
 from superannotate.logger import get_default_logger
 from tqdm import tqdm
 
@@ -1634,7 +1635,10 @@ def create_annotation_classes_from_classes_json(
         else:
             data = open(classes_json)
         classes_json = json.load(data)
-    annotation_classes = parse_obj_as(List[AnnotationClassEntity], classes_json)
+    try:
+        annotation_classes = parse_obj_as(List[AnnotationClassEntity], classes_json)
+    except ValidationError:
+        raise AppException("Couldn't validate annotation classes.")
     logger.info(
         "Creating annotation classes in project %s from %s.",
         project,

src/superannotate/lib/app/mixp/utils/parsers.py

@@ -1203,3 +1203,27 @@ def add_contributors_to_project(*args, **kwargs):
         "event_name": "add_contributors_to_project",
         "properties": {"User Role": user_role},
     }
+
+
+def get_annotations(*args, **kwargs):
+    project = kwargs.get("project", args[0])
+    items = kwargs.get("items", args[1])
+
+    return {
+        "event_name": "get_annotations",
+        "properties": {"Project": project, "items_count": len(items)},
+    }
+
+
+def get_annotations_per_frame(*args, **kwargs):
+    project = kwargs.get("project", args[0])
+    fps = kwargs.get("fps")
+    if not fps:
+        try:
+            fps = args[2]
+        except IndexError:
+            fps = 1
+    return {
+        "event_name": "get_annotations_per_frame",
+        "properties": {"Project": project, "fps": fps},
+    }

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

@@ -5,7 +5,6 @@
 from typing import List
 
 from lib.core.enums import ClassTypeEnum
-from lib.core.enums import ImageQuality
 from lib.core.enums import SegmentationStatus
 from superannotate_schemas.schemas.classes import AnnotationClass
 
@@ -177,11 +176,6 @@ def to_dict(self):
             "value": self.value,
         }
 
-    def serialize(self):
-        if self.attribute == "ImageQuality":
-            self.value = ImageQuality.get_value(self.value)
-        return self.to_dict()
-
 
 class WorkflowEntity(BaseEntity):
     def __init__(

src/superannotate/lib/infrastructure/repositories.py

@@ -20,6 +20,7 @@
 from lib.core.entities import UserEntity
 from lib.core.entities import WorkflowEntity
 from lib.core.enums import ClassTypeEnum
+from lib.core.enums import ImageQuality
 from lib.core.exceptions import AppException
 from lib.core.repositories import BaseManageableRepository
 from lib.core.repositories import BaseProjectRelatedManageableRepository
@@ -197,13 +198,15 @@ def delete(self, uuid: int):
         raise NotImplementedError
 
     def update(self, entity: ProjectSettingEntity):
-        res = self._service.set_project_settings(
-            self._project.uuid, self._project.team_id, [entity.serialize()]
+        if entity.attribute == "ImageQuality" and isinstance(entity.value, str):
+            entity.value = ImageQuality.get_value(entity.value)
+        self._service.set_project_settings(
+            self._project.uuid, self._project.team_id, [entity.to_dict()]
         )
         return entity
 
     @staticmethod
-    def dict2entity(data: dict):
+    def dict2entity(data: dict) -> ProjectSettingEntity:
         return ProjectSettingEntity(
             uuid=data["id"],
             project_id=data["project_id"],