Fix types

Author: midhun-pm

libs/labelbox/pyproject.toml

@@ -69,6 +69,7 @@ dev-dependencies = [
     "types-python-dateutil>=2.9.0.20240316",
     "types-requests>=2.31.0.20240311",
     "types-tqdm>=4.66.0.20240106",
+    "types-PyYAML>=6.0.12.20240311",
 ]
 
 [tool.ruff]

libs/labelbox/src/labelbox/alignerr/schema/project_domain.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import List, Optional, Dict, Any
 from labelbox.orm.db_object import Deletable, DbObject
 from labelbox.orm.model import Field
 from labelbox.pagination import PaginatedCollection
@@ -212,7 +212,10 @@ def get_by_project_id(
             project_id, limit, offset, include_archived
         )
 
-        params = {"projectId": project_id, "includeArchived": include_archived}
+        params: Dict[str, Any] = {
+            "projectId": project_id,
+            "includeArchived": include_archived,
+        }
 
         return PaginatedCollection(
             client=client,
@@ -268,12 +271,18 @@ def search(
             }
         }"""
 
-        params = {
+        # Build params dictionary with proper types for GraphQL
+        params: Dict[str, Any] = {
             "includeArchived": include_archived,
-            "searchByName": search_by_name,
-            "projectIds": project_ids,
         }
 
+        # Only add non-None values to avoid type issues
+        if search_by_name is not None:
+            params["searchByName"] = search_by_name
+        if project_ids is not None:
+            # Keep as list for GraphQL - it will be properly serialized
+            params["projectIds"] = project_ids
+
         return PaginatedCollection(
             client=client,
             query=query_str,