Merge pull request #411 from superannotateai/friday

Fixed streamed data chinks handeling

Author: VaghinakDev

src/superannotate/lib/infrastructure/controller.py

@@ -74,8 +74,9 @@ def __init__(self, config_path: str = None, token: str = None):
         self.initialize_backend_client()
 
     def retrieve_configs(self, path: Path, raise_exception=True):
+
         token, backend_url, ssl_verify = None, None, None
-        if not path.is_file() or not os.access(path, os.R_OK):
+        if not Path(path).is_file() or not os.access(path, os.R_OK):
             if raise_exception:
                 raise AppException(
                     f"SuperAnnotate config file {str(path)} not found."

tests/integration/test_df_processing.py

@@ -19,7 +19,7 @@ def folder_path(self):
         )
 
     def test_filter_instances(self):
-        df = sa.aggregate_annotations_as_df(self.folder_path,self.PROJECT_TYPE)
+        df = sa.aggregate_annotations_as_df(self.folder_path, self.PROJECT_TYPE)
         df = df[~(df.duplicated(["instanceId", "imageName"]))]
         df = df[df.duplicated(["trackingId"], False) & df["trackingId"].notnull()]
         self.assertEqual(len(df), 2)
@@ -28,6 +28,7 @@ def test_filter_instances(self):
             {"example_image_1.jpg", "example_image_2.jpg"},
         )
 
+
 class TestDFWithTagInstace(BaseTestCase):
     PROJECT_TYPE = "Vector"
     TEST_FOLDER_PATH = "data_set/sample_project_vector_with_tag"
@@ -38,13 +39,11 @@ def folder_path(self):
             Path(os.path.join(dirname(dirname(__file__)), self.TEST_FOLDER_PATH))
         )
 
-
     def test_filter_instances(self):
         df = sa.aggregate_annotations_as_df(self.folder_path, self.PROJECT_TYPE)
         self.assertEqual(df.iloc[0]["type"], "tag")
 
 
-
 class TestClassDistibutionWithTagInstance(BaseTestCase):
     PROJECT_TYPE = "Vector"
     EXPORT_ROOT_PATH = "data_set"
@@ -57,6 +56,6 @@ def root_path(self):
         )
 
     def test_filter_instances(self):
-        df = sa.class_distribution(export_root=self.root_path,project_names=[self.PROJECT_NAME])
+        df = sa.class_distribution(export_root=self.root_path, project_names=[self.PROJECT_NAME])
         self.assertEqual(df.iloc[0]['count'], 1)
-        self.assertEqual(df.iloc[0]['className'], "Weather")
+        self.assertEqual(df.iloc[0]['className'], "Weather")