Merge pull request #389 from superannotateai/friday-merge

validate

Author: dshabin

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

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

@@ -1637,7 +1637,9 @@ def create_annotation_classes_from_classes_json(
         classes_json = json.load(data)
     annotation_classes = parse_obj_as(List[AnnotationClassEntity], classes_json)
     logger.info(
-        "Creating annotation classes in project %s from %s.", project, classes_json_initial,
+        "Creating annotation classes in project %s from %s.",
+        project,
+        classes_json_initial,
     )
     response = controller.create_annotation_classes(
         project_name=project, annotation_classes=annotation_classes,

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

@@ -95,9 +95,7 @@ class ClassType(StrictStr):
     def validate(cls, value: Union[str]) -> Union[str]:
         enum_values = [e.name.lower() for e in ClassTypeEnum]
         if value.lower() not in enum_values:
-            raise TypeError(
-                f"Available class_types are {', '.join(enum_values)}. "
-            )
+            raise TypeError(f"Available class_types are {', '.join(enum_values)}. ")
         return value.lower()
 
 

src/superannotate/lib/core/data_handlers.py

@@ -43,7 +43,9 @@ def get_annotation_class(
         self, name: str, class_type: ClassTypeEnum
     ) -> AnnotationClass:
         for annotation_class in self._annotation_classes:
-            if annotation_class.name == name and class_type.equals(annotation_class.type):
+            if annotation_class.name == name and class_type.equals(
+                annotation_class.type
+            ):
                 return annotation_class
 
     @lru_cache()
@@ -207,7 +209,9 @@ def handle(self, annotation: dict):
             i for i in annotation["instances"] if "className" in i and i["classId"] > 0
         ]:
             annotation_class_name = annotation_instance["className"]
-            annotation_class_type = self._get_class_type(annotation_instance.get("type", ClassTypeEnum.OBJECT))
+            annotation_class_type = self._get_class_type(
+                annotation_instance.get("type", ClassTypeEnum.OBJECT)
+            )
 
             annotation_class = self.get_annotation_class(
                 annotation_class_name, annotation_class_type
@@ -287,7 +291,9 @@ def convert_timestamp(timestamp):
                 "locked": False,
             }
             if class_name:
-                annotation_class = self.get_annotation_class(class_name,ClassTypeEnum.OBJECT)
+                annotation_class = self.get_annotation_class(
+                    class_name, ClassTypeEnum.OBJECT
+                )
                 if annotation_class:
                     editor_instance["classId"] = annotation_class.id
                 else:
@@ -322,7 +328,9 @@ def convert_timestamp(timestamp):
                         ] = timestamp_data["points"]
                     if not class_name:
                         continue
-                    annotation_class = self.get_annotation_class(class_name,ClassTypeEnum.OBJECT)
+                    annotation_class = self.get_annotation_class(
+                        class_name, ClassTypeEnum.OBJECT
+                    )
                     if not annotation_class:
                         self.reporter.store_message(
                             "missing_classes", meta["className"]

src/superannotate/lib/core/types.py

@@ -1,14 +1,10 @@
-from typing import Dict
-from typing import List
 from typing import Optional
 from typing import Union
 
 from pydantic import BaseModel
 from pydantic import constr
 from pydantic import Extra
 from pydantic import StrictStr
-from pydantic import validate_model
-from pydantic import validator
 from pydantic.error_wrappers import ErrorWrapper
 from pydantic.error_wrappers import ValidationError
 from superannotate_schemas.schemas.classes import AttributeGroup as AttributeGroupSchema
@@ -28,7 +24,6 @@ def validate(cls, value: str) -> Union[str]:
         return value
 
 
-
 class Project(BaseModel):
     name: NotEmptyStr
 
@@ -45,5 +40,3 @@ class MLModel(BaseModel):
 
     class Config:
         extra = Extra.allow
-
-

src/superannotate/lib/infrastructure/controller.py

@@ -70,7 +70,9 @@ def __init__(self, config_path: str, token: str = None):
         self._team_name = None
         self._reporter = None
         self._ssl_verify = not os.environ.get("SA_TESTING", False)
-        self._config_path = expanduser(config_path) if config_path else constances.CONFIG_FILE_LOCATION
+        self._config_path = (
+            expanduser(config_path) if config_path else constances.CONFIG_FILE_LOCATION
+        )
 
         self._backend_url = os.environ.get("SA_URL", constances.BACKEND_URL)
         if not token and not config_path:
@@ -261,7 +263,6 @@ def annotation_validators(self) -> AnnotationValidators:
 
 
 class Controller(BaseController):
-
     def __init__(self, config_path: str = None):
         super().__init__(config_path)
         self._team = None
@@ -1150,7 +1151,12 @@ def get_duplicate_images(self, project_name: str, folder_name: str, images: list
         )
 
     def create_annotation_class(
-        self, project_name: str, name: str, color: str, attribute_groups: List[dict], class_type: str
+        self,
+        project_name: str,
+        name: str,
+        color: str,
+        attribute_groups: List[dict],
+        class_type: str,
     ):
         project = self._get_project(project_name)
         annotation_classes = AnnotationClassRepository(

src/superannotate/lib/infrastructure/repositories.py

@@ -360,7 +360,7 @@ def dict2entity(data: dict) -> AnnotationClassEntity:
             createdAt=data["createdAt"],
             updatedAt=data["updatedAt"],
             attribute_groups=data["attribute_groups"],
-            type=ClassTypeEnum.get_name(data.get('type')),
+            type=ClassTypeEnum.get_name(data.get("type")),
         )
 
 

src/superannotate/logger.py

@@ -43,9 +43,11 @@ def get_default_logger():
                     expanduser(constances.LOG_FILE_LOCATION),
                     maxBytes=5 * 1024 * 1024,
                     backupCount=5,
-                    mode="a"
+                    mode="a",
+                )
+                formatter = Formatter(
+                    "SA-PYTHON-SDK - %(levelname)s - %(asctime)s - %(message)s"
                 )
-                formatter = Formatter("SA-PYTHON-SDK - %(levelname)s - %(asctime)s - %(message)s")
                 file_handler.setFormatter(formatter)
                 file_handler.setLevel("DEBUG")
                 default_logger.addHandler(file_handler)