Fix stage filter

Val Brodsky · Val Brodsky · commit 94f85ff56902 · 2024-08-23T15:56:43.000-07:00
diff --git a/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py b/libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py
@@ -167,7 +167,6 @@ def get_all(
             search_query=build_search_filter(search_query)
             if search_query else None,
         )
-
         params: Dict[str, Union[str, int]] = {}
 
         def convert_to_labeling_service_dashboard(client, data):
diff --git a/libs/labelbox/src/labelbox/schema/search_filters.py b/libs/labelbox/src/labelbox/schema/search_filters.py
@@ -3,6 +3,7 @@
 from typing import List, Literal, Union
 
 from labelbox.pydantic_compat import BaseModel
+from labelbox.schema.labeling_service_status import LabelingServiceStatus
 from labelbox.utils import format_iso_datetime
 
 
@@ -108,7 +109,7 @@ class ProjectStageFilter(BaseSearchFilter):
     """
     operation: Literal[OperationType.Stage] = OperationType.Stage
     operator: IdOperator
-    values: List[str]
+    values: List[LabelingServiceStatus]
 
 
 class DateValue(BaseSearchFilter):

Original file line number	Diff line number	Diff line change
`@@ -167,7 +167,6 @@ def get_all(`
`167`	`167`	`search_query=build_search_filter(search_query)`
`168`	`168`	`if search_query else None,`
`169`	`169`	`)`
`170`		`-`
`171`	`170`	`params: Dict[str, Union[str, int]] = {}`
`172`	`171`
`173`	`172`	`def convert_to_labeling_service_dashboard(client, data):`