round frames count in VideoFrameGenerator

Author: VaghinakDev

src/superannotate/lib/core/video_convertor.py

@@ -42,7 +42,7 @@ def __init__(self, annotation_data: dict, fps: int):
         self.fps = fps
         self.ratio = 1000 * 1000 / fps
         self._frame_id = 1
-        self.frames_count = self.duration * fps
+        self.frames_count = int(self.duration * fps)
         self.annotations: dict = {}
         self._mapping = {}
         self._process()

tests/integration/annotations/test_uopload_annotations_without_classes.py

@@ -37,7 +37,7 @@ def test_annotation_upload(self):
             with tempfile.TemporaryDirectory() as classes_dir:
                 classes_path = sa.download_annotation_classes_json(self.PROJECT_NAME, classes_dir)
                 self.assertEqual(json.load(open(classes_path, "r")), [])
-                sa.create_annotation_class(self.PROJECT_NAME, "tt", "#FFFFFF", type="tag")
+                sa.create_annotation_class(self.PROJECT_NAME, "tt", "#FFFFFF", class_type="tag")
                 classes_path = sa.download_annotation_classes_json(self.PROJECT_NAME, classes_dir)
                 classes_json = json.load(open(classes_path, "r"))
                 self.assertEqual(classes_json[0]["type"], "tag")

tests/integration/classes/test_create_annotation_class.py

@@ -19,7 +19,7 @@ def large_json_path(self):
         return os.path.join(DATA_SET_PATH, self.TEST_LARGE_CLASSES_JSON)
 
     def test_create_annotation_class(self):
-        sa.create_annotation_class(self.PROJECT_NAME, "test_add", "#FF0000", type="tag")
+        sa.create_annotation_class(self.PROJECT_NAME, "test_add", "#FF0000", class_type="tag")
         classes = sa.search_annotation_classes(self.PROJECT_NAME)
         self.assertEqual(classes[0]["type"], "tag")
 
@@ -36,7 +36,7 @@ class TestCreateAnnotationClassNonVectorWithError(BaseTestCase):
     def test_create_annotation_class(self):
         msg = ""
         try:
-            sa.create_annotation_class(self.PROJECT_NAME, "test_add", "#FF0000", type="tag")
+            sa.create_annotation_class(self.PROJECT_NAME, "test_add", "#FF0000", class_type="tag")
         except Exception as e:
             msg = str(e)
         self.assertEqual(msg, "Predefined tagging functionality is not supported for projects of type Video.")

tests/integration/test_depricated_functions_document.py

@@ -12,7 +12,7 @@
 
 
 class TestDeprecatedFunctionsDocument(TestCase):
-    PROJECT_NAME = "document proj 11"
+    PROJECT_NAME = "TestDeprecatedFunctionsDocument"
     PROJECT_DESCRIPTION = "desc"
     PROJECT_TYPE = "Document"
     PATH_TO_URLS = "data_set/attach_urls.csv"

tests/integration/test_depricated_functions_video.py

@@ -12,7 +12,7 @@
 
 
 class TestDeprecatedFunctionsVideo(TestCase):
-    PROJECT_NAME = "vid proj 11"
+    PROJECT_NAME = "TestDeprecatedFunctionsVideo"
     PROJECT_DESCRIPTION = "desc"
     PROJECT_TYPE = "Video"
     PATH_TO_URLS = "data_set/attach_urls.csv"

tests/integration/test_interface.py

@@ -73,7 +73,7 @@ def test_get_project_metadata(self):
         metadata = sa.get_project_metadata(self.PROJECT_NAME)
         self.assertIsNotNone(metadata["id"])
         self.assertListEqual(metadata.get("contributors", []), [])
-        sa.create_annotation_class(self.PROJECT_NAME, "tt", "#FFFFFF", type="tag")
+        sa.create_annotation_class(self.PROJECT_NAME, "tt", "#FFFFFF", class_type="tag")
         metadata_with_users = sa.get_project_metadata(self.PROJECT_NAME, include_annotation_classes=True, include_contributors=True)
         self.assertEqual(metadata_with_users['classes'][0]['type'], 'tag')
         self.assertIsNotNone(metadata_with_users.get("contributors"))