Fix lint issues

Author: Val Brodsky

libs/labelbox/src/labelbox/schema/labeling_service_dashboard.py

@@ -71,13 +71,17 @@ class LabelingServiceDashboard(_CamelCaseMixin):
     created_at: Optional[datetime] = Field(frozen=True, default=None)
     updated_at: Optional[datetime] = Field(frozen=True, default=None)
     created_by_id: Optional[str] = Field(frozen=True, default=None)
-    status: LabelingServiceStatus = Field(frozen=True, default=None)
+    status: Optional[LabelingServiceStatus] = Field(frozen=True, default=None)
     data_rows_count: int = Field(frozen=True)
     tasks_completed_count: int = Field(frozen=True)
     tasks_remaining_count: Optional[int] = Field(frozen=True, default=None)
     media_type: Optional[MediaType] = Field(frozen=True, default=None)
-    editor_task_type: EditorTaskType = Field(frozen=True, default=None)
-    tags: List[LabelingServiceDashboardTags] = Field(frozen=True, default=None)
+    editor_task_type: Optional[EditorTaskType] = Field(
+        frozen=True, default=None
+    )
+    tags: Optional[List[LabelingServiceDashboardTags]] = Field(
+        frozen=True, default=None
+    )
 
     client: Any  # type Any to avoid circular import from client
 

libs/labelbox/src/labelbox/schema/tool_building/classification.py

@@ -66,7 +66,7 @@ class UIMode(Enum):
     options: List["Option"] = field(default_factory=list)
     schema_id: Optional[str] = None
     feature_schema_id: Optional[str] = None
-    scope: Scope = None
+    scope: Optional[Scope] = None
     ui_mode: Optional[UIMode] = (
         None  # How this classification should be answered (e.g. hotkeys / autocomplete, etc)
     )
@@ -203,7 +203,7 @@ def add_option(
                 f"Duplicate nested classification '{option.name}' "
                 f"for option '{self.label}'"
             )
-        self.options.append(option)
+        self.options.append(option)  # type: ignore
 
 
 @dataclass
@@ -305,7 +305,7 @@ def asdict(self, is_subclass: bool = False) -> Dict[str, Any]:
             raise InconsistentOntologyException(
                 f"Response Classification '{self.name}' requires options."
             )
-        classification = {
+        classification: Dict[str, Any] = {
             "type": self.class_type.value,
             "instructions": self.instructions,
             "name": self.name,

libs/labelbox/src/labelbox/schema/tool_building/types.py

@@ -1,10 +1,6 @@
-from typing import Annotated, Type
+from typing import Annotated
 
-from pydantic import StringConstraints
+from pydantic import Field
 
-FeatureSchemaId: Type[str] = Annotated[
-    str, StringConstraints(min_length=25, max_length=25)
-]
-SchemaId: Type[str] = Annotated[
-    str, StringConstraints(min_length=25, max_length=25)
-]
+FeatureSchemaId = Annotated[str, Field(min_length=25, max_length=25)]
+SchemaId = Annotated[str, Field(min_length=25, max_length=25)]

libs/labelbox/tests/integration/test_chat_evaluation_ontology_project.py

@@ -15,7 +15,7 @@ def test_create_chat_evaluation_ontology_project(
     # here we are essentially testing the ontology creation which is a fixture
     assert ontology
     assert ontology.name
-    assert len(ontology.tools()) == 5
+    assert len(ontology.tools()) == 6
     for tool in ontology.tools():
         assert tool.schema_id
         assert tool.feature_schema_id

libs/labelbox/tests/unit/test_unit_fact_checking_tool.py

@@ -45,9 +45,8 @@ def test_fact_checking_as_dict_default():
 
 def test_step_reasoning_as_dict_with_actions():
     tool = FactCheckingTool(name="Fact Checking Tool")
-    tool.set_unsupported_step_actions([])
-    tool.set_cant_confidently_assess_step_actions([])
-    tool.set_no_factual_information_step_actions([])
+    for variant in tool.definition.variants:
+        variant.set_actions([])
 
     # Get the dictionary representation
     tool_dict = tool.asdict()
@@ -59,11 +58,12 @@ def test_step_reasoning_as_dict_with_actions():
         "required": False,
         "schemaNodeId": None,
         "featureSchemaId": None,
+        "color": None,
         "definition": {
             "variants": [
-                {"id": 0, "name": "Accurate"},
-                {"id": 1, "name": "Inaccurate"},
-                {"id": 2, "name": "Disputed"},
+                {"id": 0, "name": "Accurate", "actions": []},
+                {"id": 1, "name": "Inaccurate", "actions": []},
+                {"id": 2, "name": "Disputed", "actions": []},
                 {
                     "id": 3,
                     "name": "Unsupported",