Fix wm serializers

Author: VaghinakDev

pytest.ini

@@ -3,4 +3,4 @@ minversion = 3.7
 log_cli=true
 python_files = test_*.py
 ;pytest_plugins = ['pytest_profiling']
-;addopts = -n 6 --dist loadscope
+addopts = -n 6 --dist loadscope

src/superannotate/lib/core/serviceproviders.py

@@ -167,7 +167,7 @@ def create_project_categories(
     def list_users(
         self,
         body_query: Query,
-        parent_entity: str = "Team",
+        parent_entity: CustomFieldEntityEnum = CustomFieldEntityEnum.TEAM,
         chunk_size=100,
         project_id: int = None,
         include_custom_fields=False,

src/superannotate/lib/infrastructure/services/work_management.py

@@ -318,14 +318,14 @@ def list_users(
         self,
         body_query: Query,
         chunk_size=100,
-        parent_entity: str = "Team",
+        parent_entity: CustomFieldEntityEnum = CustomFieldEntityEnum.TEAM,
         project_id: int = None,
         include_custom_fields=False,
     ) -> WMUserListResponse:
         if include_custom_fields:
             url = self.URL_SEARCH_CUSTOM_ENTITIES
         else:
-            if parent_entity == "Team":
+            if parent_entity == CustomFieldEntityEnum.TEAM:
                 url = self.URL_SEARCH_TEAM_USERS
             else:
                 url = self.URL_SEARCH_PROJECT_USERS
@@ -344,7 +344,7 @@ def list_users(
             body_query=body_query,
             query_params={
                 "entity": "Contributor",
-                "parentEntity": parent_entity,
+                "parentEntity": parent_entity.value,
             },
             headers={
                 "x-sa-entity-context": entity_context,