Update data typing

Author: VaghinakDev

CHANGELOG.rst

@@ -11,12 +11,12 @@ ______________________
 
 **Added**
 
-    - ``SAClient.create_categories`` creates one or more categories in a project. 
+    - ``SAClient.create_categories`` creates one or more categories in a project.
     - ``SAClient.remove_categories`` removes one or more categories in a project.
     - ``SAClient.list_categories`` lists all categories in the project.
     - ``SAClient.set_contributors_categories`` assigns one or more categories to specified contributors.
     - ``SAClient.remove_contributors_categories`` removes specified categories from contributors.
-    - ``SAClient.set_items_category`` adds categories to one or more items. 
+    - ``SAClient.set_items_category`` adds categories to one or more items.
     - ``SAClient.remove_items_category`` removes categories from one or more items.
 
 **Updated**

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

@@ -2,6 +2,7 @@
 from enum import auto
 from enum import Enum
 from typing import Any
+from typing import List
 from typing import Optional
 from typing import Union
 
@@ -134,7 +135,7 @@ class WMProjectUserEntity(TimedBaseModel):
     state: Optional[WMUserStateEnum]
     custom_fields: Optional[dict] = Field(dict(), alias="customField")
     permissions: Optional[dict]
-    categories: Optional[list[dict]]
+    categories: Optional[List[dict]]
 
     class Config:
         extra = Extra.ignore

src/superannotate/lib/core/serviceproviders.py

@@ -236,7 +236,7 @@ def set_remove_contributor_categories(
         category_ids: List[int],
         operation: Literal["set", "remove"],
         chunk_size=100,
-    ) -> list[dict]:
+    ) -> List[dict]:
         raise NotImplementedError
 
 

tests/integration/items/test_attach_category.py

@@ -2,6 +2,7 @@
 import os
 import time
 from pathlib import Path
+from typing import List
 from unittest import TestCase
 
 from src.superannotate import SAClient
@@ -59,7 +60,7 @@ def tearDownClass(cls) -> None:
                 pass
 
     @staticmethod
-    def _attach_items(path: str, names: list[str]):
+    def _attach_items(path: str, names: List[str]):
         sa.attach_items(path, [{"name": name, "url": f"url-{name}"} for name in names])
 
     def test_attache_category(self):

tests/integration/items/test_list_items.py

@@ -109,7 +109,9 @@ def test_list_category_filter(self):
             include=["categories"],
             categories__value__in=["c1", "c2"],
         )
-        assert [i["categories"][0]["value"] for i in items] == ["c1", "c2"]
+        assert sorted([i["categories"][0]["value"] for i in items]) == sorted(
+            ["c1", "c2"]
+        )
         assert (
             len(
                 sa.list_items(