Skip to content

Commit 9ea76fd

Browse files
author
Val Brodsky
committed
Move all files from ontology_building to tool_building
1 parent cd59a76 commit 9ea76fd

File tree

16 files changed

+27
-136
lines changed

16 files changed

+27
-136
lines changed

libs/labelbox/src/labelbox/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@
5555
)
5656
from labelbox.schema.tool_building.fact_checking_tool import FactCheckingTool
5757
from labelbox.schema.tool_building.step_reasoning_tool import StepReasoningTool
58+
from labelbox.schema.tool_building.prompt_issue_tool import PromptIssueTool
5859
from labelbox.schema.role import Role, ProjectRole
5960
from labelbox.schema.invite import Invite, InviteLimit
6061
from labelbox.schema.data_row_metadata import (
@@ -92,7 +93,7 @@
9293
from labelbox.schema.task_queue import TaskQueue
9394
from labelbox.schema.user import User
9495
from labelbox.schema.webhook import Webhook
95-
from labelbox.schema.ontology_building.classification import (
96+
from labelbox.schema.tool_building.classification import (
9697
Classification,
9798
Option,
9899
ResponseOption,

libs/labelbox/src/labelbox/schema/data_row_metadata.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929

3030
from labelbox.schema.identifiable import GlobalKey, UniqueId
3131
from labelbox.schema.identifiables import DataRowIdentifiers, UniqueIds
32-
from labelbox.schema.ontology_building.types import SchemaId
32+
from labelbox.schema.tool_building.types import SchemaId
3333
from labelbox.utils import (
3434
_CamelCaseMixin,
3535
format_iso_datetime,

libs/labelbox/src/labelbox/schema/ontology.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,19 +10,19 @@
1010

1111
from labelbox.orm.db_object import DbObject
1212
from labelbox.orm.model import Field, Relationship
13-
from labelbox.schema.ontology_building.classification import (
13+
from labelbox.schema.ontology_building.prompt_issue_tool import PromptIssueTool
14+
from labelbox.schema.tool_building.classification import (
1415
Classification,
1516
PromptResponseClassification,
1617
)
17-
from labelbox.schema.ontology_building.fact_checking_tool import (
18+
from labelbox.schema.tool_building.fact_checking_tool import (
1819
FactCheckingTool,
1920
)
20-
from labelbox.schema.ontology_building.prompt_issue_tool import PromptIssueTool
21-
from labelbox.schema.ontology_building.step_reasoning_tool import (
21+
from labelbox.schema.tool_building.step_reasoning_tool import (
2222
StepReasoningTool,
2323
)
24-
from labelbox.schema.ontology_building.tool_type import ToolType
25-
from labelbox.schema.ontology_building.tool_type_mapping import (
24+
from labelbox.schema.tool_building.tool_type import ToolType
25+
from labelbox.schema.tool_building.tool_type_mapping import (
2626
map_tool_type_to_tool_cls,
2727
)
2828

libs/labelbox/src/labelbox/schema/ontology_building/__init__.py

Lines changed: 0 additions & 7 deletions
This file was deleted.

libs/labelbox/src/labelbox/schema/ontology_building/prompt_issue_tool.py

Lines changed: 0 additions & 77 deletions
This file was deleted.

libs/labelbox/src/labelbox/schema/ontology_building/variant.py

Lines changed: 0 additions & 36 deletions
This file was deleted.
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
import labelbox.schema.tool_building.tool_type
2+
import labelbox.schema.ontology_building.variant
3+
import labelbox.schema.tool_building.step_reasoning_tool
4+
import labelbox.schema.tool_building.fact_checking_tool
5+
import labelbox.schema.tool_building.prompt_issue_tool
6+
import labelbox.schema.tool_building.tool_type_mapping
7+
import labelbox.schema.tool_building.types
8+
import labelbox.schema.tool_building.classification

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
from dataclasses import dataclass, field
22
from typing import Any, Dict, List, Optional
33

4-
from labelbox.schema.ontology_building.classification import (
4+
from labelbox.schema.tool_building.classification import (
55
Classification,
66
)
7-
from labelbox.schema.ontology_building.tool_type import ToolType
7+
from labelbox.schema.tool_building.tool_type import ToolType
88

99

1010
def _supported_classifications() -> List[Classification]:

libs/labelbox/src/labelbox/schema/ontology_building/tool_type_mapping.py renamed to libs/labelbox/src/labelbox/schema/tool_building/tool_type_mapping.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,3 +16,5 @@ def map_tool_type_to_tool_cls(tool_type_str: str):
1616
return StepReasoningTool
1717
elif tool_type == ToolType.FACT_CHECKING:
1818
return FactCheckingTool
19+
elif tool_type == ToolType.PROMPT_ISSUE:
20+
return PromptIssueTool

libs/labelbox/tests/integration/conftest.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@
2222
Tool,
2323
)
2424
from labelbox.schema.data_row import DataRowMetadataField
25-
from labelbox.schema.ontology_building.fact_checking_tool import (
25+
from labelbox.schema.ontology_building.prompt_issue_tool import PromptIssueTool
26+
from labelbox.schema.ontology_kind import OntologyKind
27+
from labelbox.schema.tool_building.fact_checking_tool import (
2628
FactCheckingTool,
2729
)
28-
from labelbox.schema.ontology_building.prompt_issue_tool import PromptIssueTool
29-
from labelbox.schema.ontology_building.step_reasoning_tool import (
30+
from labelbox.schema.tool_building.step_reasoning_tool import (
3031
StepReasoningTool,
3132
)
32-
from labelbox.schema.ontology_kind import OntologyKind
3333
from labelbox.schema.user import User
3434

3535

libs/labelbox/tests/integration/test_ontology.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,10 @@
55

66
from labelbox import MediaType, OntologyBuilder, Tool
77
from labelbox.orm.model import Entity
8-
from labelbox.schema.ontology_building.fact_checking_tool import (
8+
from labelbox.schema.tool_building.fact_checking_tool import (
99
FactCheckingTool,
1010
)
11-
from labelbox.schema.ontology_building.step_reasoning_tool import (
11+
from labelbox.schema.tool_building.step_reasoning_tool import (
1212
StepReasoningTool,
1313
)
1414

0 commit comments

Comments
 (0)