testing

Author: Val Brodsky

libs/labelbox/src/labelbox/__init__.py

@@ -92,3 +92,8 @@
 from labelbox.schema.task_queue import TaskQueue
 from labelbox.schema.user import User
 from labelbox.schema.webhook import Webhook
+from labelbox.schema.ontology_building.classification import (
+    Classification,
+    Option,
+    ResponseOption,
+)

libs/labelbox/tests/integration/conftest.py

@@ -22,6 +22,13 @@
     Tool,
 )
 from labelbox.schema.data_row import DataRowMetadataField
+from labelbox.schema.ontology_building.fact_checking_tool import (
+    FactCheckingTool,
+)
+from labelbox.schema.ontology_building.prompt_issue_tool import PromptIssueTool
+from labelbox.schema.ontology_building.step_reasoning_tool import (
+    StepReasoningTool,
+)
 from labelbox.schema.ontology_kind import OntologyKind
 from labelbox.schema.user import User
 
@@ -581,6 +588,7 @@ def chat_evaluation_ontology(client, rand_gen):
             ),
             StepReasoningTool(name="step reasoning"),
             FactCheckingTool(name="fact checking"),
+            PromptIssueTool(name="prompt issue"),
         ],
         classifications=[
             Classification(

libs/labelbox/tests/integration/test_ontology.py

@@ -5,8 +5,12 @@
 
 from labelbox import MediaType, OntologyBuilder, Tool
 from labelbox.orm.model import Entity
-from labelbox.schema.tool_building.fact_checking_tool import FactCheckingTool
-from labelbox.schema.tool_building.step_reasoning_tool import StepReasoningTool
+from labelbox.schema.ontology_building.fact_checking_tool import (
+    FactCheckingTool,
+)
+from labelbox.schema.ontology_building.step_reasoning_tool import (
+    StepReasoningTool,
+)
 
 
 def test_feature_schema_is_not_archived(client, ontology):