Test fix

Author: VaghinakDev

src/superannotate/__init__.py

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

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

@@ -99,7 +99,8 @@ def aggregate_image_annotations_as_df(
         class_name_to_color[name] = color
         class_group_name_to_values[name] = {}
         for attribute_group in annotation_class["attribute_groups"]:
-            if attribute_group["group_type"] in ["text", "numeric"]:
+            group_type = attribute_group.get("group_type")
+            if group_type and group_type in ["text", "numeric"]:
                 freestyle_attributes.add(attribute_group["name"])
             class_group_name_to_values[name][attribute_group["name"]] = []
             for attribute in attribute_group["attributes"]:

tests/integration/annotations/video/test_get_annotations_per_frame.py

@@ -2,7 +2,6 @@
 import math
 import os
 from os.path import dirname
-from pathlib import Path
 
 from src.superannotate import SAClient
 from tests.integration.base import BaseTestCase
@@ -24,7 +23,7 @@ class TestGetAnnotations(BaseTestCase):
 
     @property
     def csv_path(self):
-        return os.path.join(dirname(dirname(dirname(__file__))), self.PATH_TO_URLS)
+        return os.path.join(DATA_SET_PATH, self.PATH_TO_URLS)
 
     @property
     def classes_path(self):

tests/integration/classes/test_create_annotation_class.py

@@ -146,21 +146,6 @@ def test_create_annotation_class(self):
             msg = str(e)
         self.assertEqual(msg, "Predefined tagging functionality is not supported for projects of type Video.")
 
-    def test_create_supported_annotation_class(self):
-        msg = ""
-        try:
-            sa.create_annotation_class(
-                self.PROJECT_NAME, "test_add", "#FF0000",
-                attribute_groups=[
-                    {
-                        "group_type": "text",
-                        "name": "name",
-                    }
-                ]
-            )
-        except Exception as e:
-            msg = str(e)
-        self.assertEqual(msg, "This project type doesn't support the attribute group types 'text' and 'numeric'.")
 
     def test_create_radio_annotation_class_attr_required(self):
         msg = ""