Fix tests

Author: Val Brodsky

libs/labelbox/mypy.ini

@@ -12,5 +12,5 @@ ignore_errors = True
 [mypy-lbox.exceptions]
 ignore_missing_imports = True
 
-[mypy-lbox.call_info"]
+[mypy-lbox.call_info]
 ignore_missing_imports = True

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

@@ -1,3 +1,4 @@
+import warnings
 from dataclasses import dataclass, field
 from typing import Any, Dict, List, Optional
 
@@ -156,6 +157,11 @@ class StepReasoningTool:
         default_factory=StepReasoningDefinition
     )
 
+    def __post_init__(self):
+        warnings.warn(
+            "This feature is experimental and subject to change.",
+        )
+
     def reset_regenerate_conversations_after_incorrect_step(self):
         """
         For live models, the default acation will invoke the model to generate alternatives if a step is marked as incorrect

libs/labelbox/tests/integration/conftest.py

@@ -22,7 +22,6 @@
 from labelbox.schema.data_row import DataRowMetadataField
 from labelbox.schema.ontology_kind import OntologyKind
 from labelbox.schema.tool_building.step_reasoning_tool import StepReasoningTool
-from labelbox.schema.tool_building.tool_type import ToolType
 from labelbox.schema.user import User
 
 

libs/labelbox/tests/integration/test_ontology.py

@@ -339,7 +339,7 @@ def test_step_reasoning_ontology(chat_evaluation_ontology):
         {
             "id": 2,
             "name": "Incorrect",
-            "actions": [],
+            "actions": ["regenerateSteps"],
         },
     ]
     assert step_reasoning_tool["definition"]["version"] == 1
@@ -364,6 +364,6 @@ def test_step_reasoning_ontology(chat_evaluation_ontology):
         {
             "id": 2,
             "name": "Incorrect",
-            "actions": [],
+            "actions": ["regenerateSteps"],
         },
     ]