Add ProjectStageFilter validations

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/search_filters.py

@@ -2,7 +2,7 @@
 from enum import Enum
 from typing import List, Literal, Union
 
-from labelbox.pydantic_compat import BaseModel
+from labelbox.pydantic_compat import BaseModel, validator
 from labelbox.schema.labeling_service_status import LabelingServiceStatus
 from labelbox.utils import format_iso_datetime
 
@@ -121,6 +121,16 @@ class ProjectStageFilter(BaseSearchFilter):
     operator: IdOperator
     values: List[LabelingServiceStatus]
 
+    @validator('values', pre=True)
+    def validate_values(cls, values):
+        disallowed_values = [LabelingServiceStatus.Missing]
+        for value in values:
+            if value in disallowed_values:
+                raise ValueError(
+                    f"{value} is not a valid value for ProjectStageFilter")
+
+        return values
+
 
 class DateValue(BaseSearchFilter):
     """

libs/labelbox/tests/unit/test_unit_search_filters.py

@@ -2,6 +2,7 @@
 from labelbox.schema.labeling_service import LabelingServiceStatus
 from labelbox.schema.search_filters import IntegerValue, RangeDateTimeOperatorWithSingleValue, RangeOperatorWithSingleValue, DateRange, RangeOperatorWithValue, DateRangeValue, DateValue, IdOperator, OperationType, OrganizationFilter, ProjectStageFilter, SharedWithOrganizationFilter, TagFilter, TaskCompletedCountFilter, TaskRemainingCountFilter, WorkforceRequestedDateFilter, WorkforceRequestedDateRangeFilter, WorkforceStageUpdatedFilter, WorkforceStageUpdatedRangeFilter, WorkspaceFilter, build_search_filter
 from labelbox.utils import format_iso_datetime
+import pytest
 
 
 def test_id_filters():
@@ -22,6 +23,19 @@ def test_id_filters():
     ) == '[{operator: "is", values: ["clphb4vd7000cd2wv1ktu5cwa"], type: "organization_id"}, {operator: "is", values: ["clphb4vd7000cd2wv1ktu5cwa"], type: "shared_with_organizations"}, {operator: "is", values: ["clphb4vd7000cd2wv1ktu5cwa"], type: "workspace"}, {operator: "is", values: ["tag"], type: "tag"}, {operator: "is", values: ["REQUESTED"], type: "stage"}]'
 
 
+def test_stage_filter_with_invalid_values():
+    with pytest.raises(
+            ValueError,
+            match="is not a valid value for ProjectStageFilter") as e:
+        _ = [
+            ProjectStageFilter(operator=IdOperator.Is,
+                               values=[
+                                   LabelingServiceStatus.Requested,
+                                   LabelingServiceStatus.Missing
+                               ]),
+        ]
+
+
 def test_date_filters():
     local_time_start = datetime.strptime("2024-01-01", "%Y-%m-%d")
     local_time_end = datetime.strptime("2025-01-01", "%Y-%m-%d")