Update FactCheckingTool to match the latest api

Author: Val Brodsky

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

@@ -5,21 +5,24 @@
 
 from labelbox.schema.tool_building.tool_type import ToolType
 from labelbox.schema.tool_building.variant import (
-    Variant,
     VariantWithActions,
 )
 
 
+class FactCheckingActions(Enum):
+    WRITE_JUSTIFICATION = "justification"
+
+
 class UnsupportedStepActions(Enum):
-    WRITE_JUSTIFICATION = "writeJustification"
+    WRITE_JUSTIFICATION = "justification"
 
 
 class CanConfidentlyAssessStepActions(Enum):
-    WRITE_JUSTIFICATION = "writeJustification"
+    WRITE_JUSTIFICATION = "justification"
 
 
 class NoFactualInformationStepActions(Enum):
-    WRITE_JUSTIFICATION = "writeJustification"
+    WRITE_JUSTIFICATION = "justification"
 
 
 @dataclass
@@ -29,43 +32,53 @@ class FactCheckingVariants:
     NOTE do not change the variants directly
     """
 
-    accurate_step: Variant = field(
-        default_factory=lambda: Variant(id=0, name="Accurate")
+    accurate_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=0,
+            name="Accurate",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
-    inaccurate_step: Variant = field(
-        default_factory=lambda: Variant(id=1, name="Inaccurate")
+
+    inaccurate_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=1,
+            name="Inaccurate",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
-    disputed_step: Variant = field(
-        default_factory=lambda: Variant(id=2, name="Disputed")
+    disputed_step: VariantWithActions = field(
+        default_factory=lambda: VariantWithActions(
+            id=2,
+            name="Disputed",
+            _available_actions={action.value for action in FactCheckingActions},
+            actions=[action.value for action in FactCheckingActions],
+        )
     )
     unsupported_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=3,
             name="Unsupported",
-            _available_actions={
-                action.value for action in UnsupportedStepActions
-            },
-            actions=[UnsupportedStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
     cant_confidently_assess_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=4,
             name="Can't confidently assess",
-            _available_actions={
-                action.value for action in CanConfidentlyAssessStepActions
-            },
-            actions=[CanConfidentlyAssessStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
     no_factual_information_step: VariantWithActions = field(
         default_factory=lambda: VariantWithActions(
             id=5,
             name="No factual information",
-            _available_actions={
-                action.value for action in NoFactualInformationStepActions
-            },
-            actions=[NoFactualInformationStepActions.WRITE_JUSTIFICATION.value],
+            _available_actions=set(),
+            actions=[],
         )
     )
 
@@ -90,11 +103,11 @@ def from_dict(cls, dictionary: List[Dict[str, Any]]):
 
         for variant in dictionary:
             if variant["id"] == 0:
-                accurate_step = Variant(**variant)
+                accurate_step = VariantWithActions(**variant)
             elif variant["id"] == 1:
-                inaccurate_step = Variant(**variant)
+                inaccurate_step = VariantWithActions(**variant)
             elif variant["id"] == 2:
-                disputed_step = Variant(**variant)
+                disputed_step = VariantWithActions(**variant)
             elif variant["id"] == 3:
                 unsupported_step = VariantWithActions(**variant)
             elif variant["id"] == 4:

libs/labelbox/tests/unit/test_unit_fact_checking_tool.py

@@ -16,23 +16,23 @@ def test_fact_checking_as_dict_default():
         "featureSchemaId": None,
         "definition": {
             "variants": [
-                {"id": 0, "name": "Accurate"},
-                {"id": 1, "name": "Inaccurate"},
-                {"id": 2, "name": "Disputed"},
+                {"id": 0, "name": "Accurate", "actions": ["justification"]},
+                {"id": 1, "name": "Inaccurate", "actions": ["justification"]},
+                {"id": 2, "name": "Disputed", "actions": ["justification"]},
                 {
                     "id": 3,
                     "name": "Unsupported",
-                    "actions": ["writeJustification"],
+                    "actions": [],
                 },
                 {
                     "id": 4,
                     "name": "Can't confidently assess",
-                    "actions": ["writeJustification"],
+                    "actions": [],
                 },
                 {
                     "id": 5,
                     "name": "No factual information",
-                    "actions": ["writeJustification"],
+                    "actions": [],
                 },
             ],
             "version": 1,
@@ -60,9 +60,9 @@ def test_step_reasoning_as_dict_with_actions():
         "featureSchemaId": None,
         "definition": {
             "variants": [
-                {"id": 0, "name": "Accurate"},
-                {"id": 1, "name": "Inaccurate"},
-                {"id": 2, "name": "Disputed"},
+                {"id": 0, "name": "Accurate", "actions": ["justification"]},
+                {"id": 1, "name": "Inaccurate", "actions": ["justification"]},
+                {"id": 2, "name": "Disputed", "actions": ["justification"]},
                 {
                     "id": 3,
                     "name": "Unsupported",