Fix lint issues

Author: Val Brodsky

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 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: Type[str] = Annotated[str, Field(min_length=25, max_length=25)]
+SchemaId: Type[str] = 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",