rapidata 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- rapidata/rapidata_client/{order/dataset → dataset}/rapidata_dataset.py +21 -9
- rapidata/rapidata_client/dataset/rapidata_validation_set.py +78 -0
- rapidata/rapidata_client/dataset/validation_rapid_parts.py +52 -0
- rapidata/rapidata_client/{order/dataset → dataset}/validation_set_builder.py +14 -53
- rapidata/rapidata_client/feature_flags/feature_flags.py +1 -1
- rapidata/rapidata_client/order/rapidata_order.py +1 -1
- rapidata/rapidata_client/order/rapidata_order_builder.py +25 -11
- rapidata/rapidata_client/rapidata_client.py +14 -4
- rapidata/rapidata_client/types/__init__.py +1 -0
- {rapidata-0.1.10.dist-info → rapidata-0.1.12.dist-info}/METADATA +1 -1
- {rapidata-0.1.10.dist-info → rapidata-0.1.12.dist-info}/RECORD +14 -11
- /rapidata/rapidata_client/{order/dataset → dataset}/__init__.py +0 -0
- {rapidata-0.1.10.dist-info → rapidata-0.1.12.dist-info}/LICENSE +0 -0
- {rapidata-0.1.10.dist-info → rapidata-0.1.12.dist-info}/WHEEL +0 -0
|
@@ -1,6 +1,5 @@
|
|
|
1
1
|
from itertools import zip_longest
|
|
2
2
|
import os
|
|
3
|
-
from typing import List
|
|
4
3
|
|
|
5
4
|
from rapidata.api_client.models.datapoint_metadata_model import DatapointMetadataModel
|
|
6
5
|
from rapidata.api_client.models.datapoint_metadata_model_metadata_inner import (
|
|
@@ -10,6 +9,7 @@ from rapidata.api_client.models.upload_text_sources_to_dataset_model import (
|
|
|
10
9
|
UploadTextSourcesToDatasetModel,
|
|
11
10
|
)
|
|
12
11
|
from rapidata.rapidata_client.metadata.base_metadata import Metadata
|
|
12
|
+
from rapidata.rapidata_client.types import RapidAsset
|
|
13
13
|
from rapidata.service import LocalFileService
|
|
14
14
|
from rapidata.service.openapi_service import OpenAPIService
|
|
15
15
|
|
|
@@ -31,22 +31,34 @@ class RapidataDataset:
|
|
|
31
31
|
|
|
32
32
|
def add_media_from_paths(
|
|
33
33
|
self,
|
|
34
|
-
image_paths: list[
|
|
35
|
-
metadata:
|
|
34
|
+
image_paths: list[RapidAsset],
|
|
35
|
+
metadata: list[Metadata] | None = None,
|
|
36
36
|
):
|
|
37
37
|
if metadata is not None and len(metadata) != len(image_paths):
|
|
38
38
|
raise ValueError(
|
|
39
39
|
"metadata must be None or have the same length as image_paths"
|
|
40
40
|
)
|
|
41
41
|
|
|
42
|
-
for
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
42
|
+
for media_paths_rapid, meta in zip_longest(image_paths, metadata or []):
|
|
43
|
+
|
|
44
|
+
if isinstance(media_paths_rapid, list) and not all(
|
|
45
|
+
os.path.exists(media_path) for media_path in media_paths_rapid
|
|
46
|
+
):
|
|
47
|
+
raise FileNotFoundError(f"File not found: {media_paths_rapid}")
|
|
48
|
+
|
|
49
|
+
elif isinstance(media_paths_rapid, str) and not os.path.exists(
|
|
50
|
+
media_paths_rapid
|
|
51
|
+
):
|
|
52
|
+
raise FileNotFoundError(f"File not found: {media_paths_rapid}")
|
|
53
|
+
|
|
46
54
|
meta_model = meta.to_model() if meta else None
|
|
47
55
|
model = DatapointMetadataModel(
|
|
48
56
|
datasetId=self.dataset_id,
|
|
49
|
-
metadata=
|
|
57
|
+
metadata=(
|
|
58
|
+
[DatapointMetadataModelMetadataInner(meta_model)]
|
|
59
|
+
if meta_model
|
|
60
|
+
else []
|
|
61
|
+
),
|
|
50
62
|
)
|
|
51
63
|
|
|
52
|
-
self.openapi_service.dataset_api.dataset_create_datapoint_post(model=model, files=[
|
|
64
|
+
self.openapi_service.dataset_api.dataset_create_datapoint_post(model=model, files=media_paths_rapid if isinstance(media_paths_rapid, list) else [media_paths_rapid]) # type: ignore
|
|
@@ -0,0 +1,78 @@
|
|
|
1
|
+
from typing import Any
|
|
2
|
+
from rapidata.api_client.models.add_validation_rapid_model import (
|
|
3
|
+
AddValidationRapidModel,
|
|
4
|
+
)
|
|
5
|
+
from rapidata.api_client.models.add_validation_rapid_model_payload import (
|
|
6
|
+
AddValidationRapidModelPayload,
|
|
7
|
+
)
|
|
8
|
+
from rapidata.api_client.models.add_validation_rapid_model_truth import (
|
|
9
|
+
AddValidationRapidModelTruth,
|
|
10
|
+
)
|
|
11
|
+
from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
|
|
12
|
+
from rapidata.api_client.models.bounding_box_payload import BoundingBoxPayload
|
|
13
|
+
from rapidata.api_client.models.bounding_box_truth import BoundingBoxTruth
|
|
14
|
+
from rapidata.api_client.models.classify_payload import ClassifyPayload
|
|
15
|
+
from rapidata.api_client.models.compare_payload import ComparePayload
|
|
16
|
+
from rapidata.api_client.models.compare_truth import CompareTruth
|
|
17
|
+
from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
|
|
18
|
+
from rapidata.api_client.models.free_text_payload import FreeTextPayload
|
|
19
|
+
from rapidata.api_client.models.line_payload import LinePayload
|
|
20
|
+
from rapidata.api_client.models.line_truth import LineTruth
|
|
21
|
+
from rapidata.api_client.models.locate_box_truth import LocateBoxTruth
|
|
22
|
+
from rapidata.api_client.models.locate_payload import LocatePayload
|
|
23
|
+
from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
|
|
24
|
+
from rapidata.api_client.models.named_entity_truth import NamedEntityTruth
|
|
25
|
+
from rapidata.api_client.models.polygon_payload import PolygonPayload
|
|
26
|
+
from rapidata.api_client.models.polygon_truth import PolygonTruth
|
|
27
|
+
from rapidata.api_client.models.transcription_payload import TranscriptionPayload
|
|
28
|
+
from rapidata.api_client.models.transcription_truth import TranscriptionTruth
|
|
29
|
+
from rapidata.rapidata_client.dataset.validation_rapid_parts import ValidatioRapidParts
|
|
30
|
+
from rapidata.rapidata_client.types import RapidAsset
|
|
31
|
+
from rapidata.service.openapi_service import OpenAPIService
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
class RapidataValidationSet:
|
|
35
|
+
|
|
36
|
+
def __init__(self, validation_set_id, openapi_service: OpenAPIService):
|
|
37
|
+
self.id = validation_set_id
|
|
38
|
+
self.openapi_service = openapi_service
|
|
39
|
+
|
|
40
|
+
def add_validation_rapid(
|
|
41
|
+
self,
|
|
42
|
+
payload: (
|
|
43
|
+
BoundingBoxPayload
|
|
44
|
+
| ClassifyPayload
|
|
45
|
+
| ComparePayload
|
|
46
|
+
| FreeTextPayload
|
|
47
|
+
| LinePayload
|
|
48
|
+
| LocatePayload
|
|
49
|
+
| NamedEntityPayload
|
|
50
|
+
| PolygonPayload
|
|
51
|
+
| TranscriptionPayload
|
|
52
|
+
),
|
|
53
|
+
truths: (
|
|
54
|
+
AttachCategoryTruth
|
|
55
|
+
| BoundingBoxTruth
|
|
56
|
+
| CompareTruth
|
|
57
|
+
| EmptyValidationTruth
|
|
58
|
+
| LineTruth
|
|
59
|
+
| LocateBoxTruth
|
|
60
|
+
| NamedEntityTruth
|
|
61
|
+
| PolygonTruth
|
|
62
|
+
| TranscriptionTruth
|
|
63
|
+
),
|
|
64
|
+
metadata: Any,
|
|
65
|
+
media_paths: RapidAsset,
|
|
66
|
+
randomCorrectProbability: float,
|
|
67
|
+
):
|
|
68
|
+
model = AddValidationRapidModel(
|
|
69
|
+
validationSetId=self.id,
|
|
70
|
+
payload=AddValidationRapidModelPayload(payload),
|
|
71
|
+
truth=AddValidationRapidModelTruth(truths),
|
|
72
|
+
metadata=metadata or [],
|
|
73
|
+
randomCorrectProbability=randomCorrectProbability,
|
|
74
|
+
)
|
|
75
|
+
|
|
76
|
+
self.openapi_service.validation_api.validation_add_validation_rapid_post(
|
|
77
|
+
model=model, files=media_paths if isinstance(media_paths, list) else [media_paths] # type: ignore
|
|
78
|
+
)
|
|
@@ -0,0 +1,52 @@
|
|
|
1
|
+
from dataclasses import dataclass
|
|
2
|
+
from typing import Any
|
|
3
|
+
|
|
4
|
+
from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
|
|
5
|
+
from rapidata.api_client.models.bounding_box_payload import BoundingBoxPayload
|
|
6
|
+
from rapidata.api_client.models.bounding_box_truth import BoundingBoxTruth
|
|
7
|
+
from rapidata.api_client.models.classify_payload import ClassifyPayload
|
|
8
|
+
from rapidata.api_client.models.compare_payload import ComparePayload
|
|
9
|
+
from rapidata.api_client.models.compare_truth import CompareTruth
|
|
10
|
+
from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
|
|
11
|
+
from rapidata.api_client.models.free_text_payload import FreeTextPayload
|
|
12
|
+
from rapidata.api_client.models.line_payload import LinePayload
|
|
13
|
+
from rapidata.api_client.models.line_truth import LineTruth
|
|
14
|
+
from rapidata.api_client.models.locate_box_truth import LocateBoxTruth
|
|
15
|
+
from rapidata.api_client.models.locate_payload import LocatePayload
|
|
16
|
+
from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
|
|
17
|
+
from rapidata.api_client.models.named_entity_truth import NamedEntityTruth
|
|
18
|
+
from rapidata.api_client.models.polygon_payload import PolygonPayload
|
|
19
|
+
from rapidata.api_client.models.polygon_truth import PolygonTruth
|
|
20
|
+
from rapidata.api_client.models.transcription_payload import TranscriptionPayload
|
|
21
|
+
from rapidata.api_client.models.transcription_truth import TranscriptionTruth
|
|
22
|
+
from rapidata.rapidata_client.types import RapidAsset
|
|
23
|
+
|
|
24
|
+
|
|
25
|
+
@dataclass
|
|
26
|
+
class ValidatioRapidParts:
|
|
27
|
+
question: str
|
|
28
|
+
media_paths: RapidAsset
|
|
29
|
+
payload: (
|
|
30
|
+
BoundingBoxPayload
|
|
31
|
+
| ClassifyPayload
|
|
32
|
+
| ComparePayload
|
|
33
|
+
| FreeTextPayload
|
|
34
|
+
| LinePayload
|
|
35
|
+
| LocatePayload
|
|
36
|
+
| NamedEntityPayload
|
|
37
|
+
| PolygonPayload
|
|
38
|
+
| TranscriptionPayload
|
|
39
|
+
)
|
|
40
|
+
truths: (
|
|
41
|
+
AttachCategoryTruth
|
|
42
|
+
| BoundingBoxTruth
|
|
43
|
+
| CompareTruth
|
|
44
|
+
| EmptyValidationTruth
|
|
45
|
+
| LineTruth
|
|
46
|
+
| LocateBoxTruth
|
|
47
|
+
| NamedEntityTruth
|
|
48
|
+
| PolygonTruth
|
|
49
|
+
| TranscriptionTruth
|
|
50
|
+
)
|
|
51
|
+
metadata: Any
|
|
52
|
+
randomCorrectProbability: float
|
|
@@ -9,55 +9,18 @@ from rapidata.api_client.models.add_validation_rapid_model_truth import (
|
|
|
9
9
|
AddValidationRapidModelTruth,
|
|
10
10
|
)
|
|
11
11
|
from rapidata.api_client.models.attach_category_truth import AttachCategoryTruth
|
|
12
|
-
from rapidata.api_client.models.bounding_box_payload import BoundingBoxPayload
|
|
13
|
-
from rapidata.api_client.models.bounding_box_truth import BoundingBoxTruth
|
|
14
12
|
from rapidata.api_client.models.classify_payload import ClassifyPayload
|
|
15
13
|
from rapidata.api_client.models.compare_payload import ComparePayload
|
|
16
14
|
from rapidata.api_client.models.compare_truth import CompareTruth
|
|
17
|
-
from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
|
|
18
|
-
from rapidata.api_client.models.free_text_payload import FreeTextPayload
|
|
19
|
-
from rapidata.api_client.models.line_payload import LinePayload
|
|
20
|
-
from rapidata.api_client.models.line_truth import LineTruth
|
|
21
|
-
from rapidata.api_client.models.locate_box_truth import LocateBoxTruth
|
|
22
|
-
from rapidata.api_client.models.locate_payload import LocatePayload
|
|
23
|
-
from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
|
|
24
|
-
from rapidata.api_client.models.named_entity_truth import NamedEntityTruth
|
|
25
|
-
from rapidata.api_client.models.polygon_payload import PolygonPayload
|
|
26
|
-
from rapidata.api_client.models.polygon_truth import PolygonTruth
|
|
27
15
|
from rapidata.api_client.models.transcription_payload import TranscriptionPayload
|
|
28
16
|
from rapidata.api_client.models.transcription_truth import TranscriptionTruth
|
|
29
17
|
from rapidata.api_client.models.transcription_word import TranscriptionWord
|
|
18
|
+
from rapidata.rapidata_client.dataset.rapidata_validation_set import RapidataValidationSet
|
|
19
|
+
from rapidata.rapidata_client.dataset.validation_rapid_parts import ValidatioRapidParts
|
|
30
20
|
from rapidata.service.openapi_service import OpenAPIService
|
|
31
21
|
|
|
32
22
|
|
|
33
|
-
|
|
34
|
-
class ValidatioRapidParts:
|
|
35
|
-
question: str
|
|
36
|
-
media_paths: str | list[str]
|
|
37
|
-
payload: Union[
|
|
38
|
-
BoundingBoxPayload,
|
|
39
|
-
ClassifyPayload,
|
|
40
|
-
ComparePayload,
|
|
41
|
-
FreeTextPayload,
|
|
42
|
-
LinePayload,
|
|
43
|
-
LocatePayload,
|
|
44
|
-
NamedEntityPayload,
|
|
45
|
-
PolygonPayload,
|
|
46
|
-
TranscriptionPayload,
|
|
47
|
-
]
|
|
48
|
-
truths: Union[
|
|
49
|
-
AttachCategoryTruth,
|
|
50
|
-
BoundingBoxTruth,
|
|
51
|
-
CompareTruth,
|
|
52
|
-
EmptyValidationTruth,
|
|
53
|
-
LineTruth,
|
|
54
|
-
LocateBoxTruth,
|
|
55
|
-
NamedEntityTruth,
|
|
56
|
-
PolygonTruth,
|
|
57
|
-
TranscriptionTruth,
|
|
58
|
-
]
|
|
59
|
-
metadata: Any
|
|
60
|
-
randomCorrectProbability: float
|
|
23
|
+
|
|
61
24
|
|
|
62
25
|
|
|
63
26
|
class ValidationSetBuilder:
|
|
@@ -79,20 +42,18 @@ class ValidationSetBuilder:
|
|
|
79
42
|
if self.validation_set_id is None:
|
|
80
43
|
raise ValueError("Failed to create validation set")
|
|
81
44
|
|
|
45
|
+
validation_set = RapidataValidationSet(validation_set_id=self.validation_set_id, openapi_service=self.openapi_service)
|
|
46
|
+
|
|
82
47
|
for rapid_part in self._rapid_parts:
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
48
|
+
validation_set.add_validation_rapid(
|
|
49
|
+
payload=rapid_part.payload,
|
|
50
|
+
truths=rapid_part.truths,
|
|
51
|
+
metadata=rapid_part.metadata,
|
|
52
|
+
media_paths=rapid_part.media_paths,
|
|
88
53
|
randomCorrectProbability=rapid_part.randomCorrectProbability,
|
|
89
54
|
)
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
model=model, files=rapid_part.media_paths if isinstance(rapid_part.media_paths, list) else [rapid_part.media_paths] # type: ignore
|
|
93
|
-
)
|
|
94
|
-
|
|
95
|
-
return str(self.validation_set_id)
|
|
55
|
+
|
|
56
|
+
return validation_set
|
|
96
57
|
|
|
97
58
|
def add_classify_rapid(
|
|
98
59
|
self, media_path: str, question: str, categories: list[str], truths: list[str]
|
|
@@ -119,6 +80,8 @@ class ValidationSetBuilder:
|
|
|
119
80
|
|
|
120
81
|
def add_compare_rapid(self, media_paths: list[str], question: str, truth: str):
|
|
121
82
|
payload = ComparePayload(_t="ComparePayload", criteria=question)
|
|
83
|
+
# take only last part of truth path
|
|
84
|
+
truth = os.path.basename(truth)
|
|
122
85
|
model_truth = CompareTruth(_t="CompareTruth", winnerId=truth)
|
|
123
86
|
|
|
124
87
|
if len(media_paths) != 2:
|
|
@@ -129,8 +92,6 @@ class ValidationSetBuilder:
|
|
|
129
92
|
if not os.path.exists(media_path):
|
|
130
93
|
raise FileNotFoundError(f"File not found: {media_path}")
|
|
131
94
|
|
|
132
|
-
# take only last part of truth path
|
|
133
|
-
truth = os.path.basename(truth)
|
|
134
95
|
|
|
135
96
|
self._rapid_parts.append(
|
|
136
97
|
ValidatioRapidParts(
|
|
@@ -9,7 +9,7 @@ class FeatureFlags:
|
|
|
9
9
|
return [FeatureFlagModel(key=name, value=value) for name, value in self._flags.items()]
|
|
10
10
|
|
|
11
11
|
def alert_on_fast_response(self, value: int):
|
|
12
|
-
self._flags["
|
|
12
|
+
self._flags["alert_on_fast_response"] = str(value)
|
|
13
13
|
return self
|
|
14
14
|
|
|
15
15
|
def disable_translation(self, value: bool = True):
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import time
|
|
2
2
|
from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
|
|
3
3
|
from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
|
|
4
|
-
from rapidata.rapidata_client.
|
|
4
|
+
from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
|
|
5
5
|
from rapidata.rapidata_client.workflow import Workflow
|
|
6
6
|
from rapidata.api_client.models.create_order_model import CreateOrderModel
|
|
7
7
|
from rapidata.rapidata_client.referee import Referee
|
|
@@ -1,18 +1,21 @@
|
|
|
1
|
-
from rapidata.api_client.api_client import ApiClient
|
|
2
1
|
from rapidata.api_client.models.aggregator_type import AggregatorType
|
|
3
2
|
from rapidata.api_client.models.create_order_model import CreateOrderModel
|
|
4
|
-
from rapidata.api_client.models.create_order_model_referee import
|
|
5
|
-
|
|
3
|
+
from rapidata.api_client.models.create_order_model_referee import (
|
|
4
|
+
CreateOrderModelReferee,
|
|
5
|
+
)
|
|
6
|
+
from rapidata.api_client.models.create_order_model_workflow import (
|
|
7
|
+
CreateOrderModelWorkflow,
|
|
8
|
+
)
|
|
6
9
|
from rapidata.rapidata_client.feature_flags import FeatureFlags
|
|
7
10
|
from rapidata.rapidata_client.metadata.base_metadata import Metadata
|
|
8
|
-
from rapidata.rapidata_client.
|
|
11
|
+
from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
|
|
9
12
|
from rapidata.rapidata_client.referee.naive_referee import NaiveReferee
|
|
13
|
+
from rapidata.rapidata_client.types import RapidAsset
|
|
10
14
|
from rapidata.rapidata_client.workflow import Workflow
|
|
11
15
|
from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
|
|
12
16
|
from rapidata.rapidata_client.referee import Referee
|
|
13
17
|
from rapidata.service.openapi_service import OpenAPIService
|
|
14
18
|
|
|
15
|
-
|
|
16
19
|
class RapidataOrderBuilder:
|
|
17
20
|
"""
|
|
18
21
|
Builder object for creating Rapidata orders.
|
|
@@ -34,8 +37,8 @@ class RapidataOrderBuilder:
|
|
|
34
37
|
self._openapi_service = openapi_service
|
|
35
38
|
self._workflow: Workflow | None = None
|
|
36
39
|
self._referee: Referee | None = None
|
|
37
|
-
self._media_paths: list[
|
|
38
|
-
self._metadata: list[Metadata] | None = None
|
|
40
|
+
self._media_paths: list[RapidAsset] = []
|
|
41
|
+
self._metadata: list[Metadata] | None = None
|
|
39
42
|
self._aggregator: AggregatorType | None = None
|
|
40
43
|
self._validation_set_id: str | None = None
|
|
41
44
|
self._feature_flags: FeatureFlags | None = None
|
|
@@ -61,7 +64,11 @@ class RapidataOrderBuilder:
|
|
|
61
64
|
userFilters=[],
|
|
62
65
|
referee=CreateOrderModelReferee(self._referee.to_model()),
|
|
63
66
|
validationSetId=self._validation_set_id,
|
|
64
|
-
featureFlags=
|
|
67
|
+
featureFlags=(
|
|
68
|
+
self._feature_flags.to_list()
|
|
69
|
+
if self._feature_flags is not None
|
|
70
|
+
else None
|
|
71
|
+
),
|
|
65
72
|
)
|
|
66
73
|
|
|
67
74
|
result = self._openapi_service.order_api.order_create_post(
|
|
@@ -70,7 +77,11 @@ class RapidataOrderBuilder:
|
|
|
70
77
|
|
|
71
78
|
self.order_id = result.order_id
|
|
72
79
|
self._dataset = RapidataDataset(result.dataset_id, self._openapi_service)
|
|
73
|
-
order = RapidataOrder(
|
|
80
|
+
order = RapidataOrder(
|
|
81
|
+
order_id=self.order_id,
|
|
82
|
+
dataset=self._dataset,
|
|
83
|
+
openapi_service=self._openapi_service,
|
|
84
|
+
)
|
|
74
85
|
|
|
75
86
|
order.dataset.add_media_from_paths(self._media_paths, self._metadata)
|
|
76
87
|
|
|
@@ -104,7 +115,10 @@ class RapidataOrderBuilder:
|
|
|
104
115
|
return self
|
|
105
116
|
|
|
106
117
|
def media(
|
|
107
|
-
self,
|
|
118
|
+
self,
|
|
119
|
+
media_paths: list[RapidAsset],
|
|
120
|
+
metadata: list[Metadata] | None = None,
|
|
121
|
+
):
|
|
108
122
|
"""
|
|
109
123
|
Set the media assets for the order.
|
|
110
124
|
|
|
@@ -157,7 +171,7 @@ class RapidataOrderBuilder:
|
|
|
157
171
|
"""
|
|
158
172
|
self._aggregator = aggregator
|
|
159
173
|
return self
|
|
160
|
-
|
|
174
|
+
|
|
161
175
|
def validation_set(self, validation_set_id: str):
|
|
162
176
|
"""
|
|
163
177
|
Set the validation set for the order.
|
|
@@ -1,7 +1,11 @@
|
|
|
1
|
-
from rapidata.rapidata_client.
|
|
1
|
+
from rapidata.rapidata_client.dataset.rapidata_validation_set import (
|
|
2
|
+
RapidataValidationSet,
|
|
3
|
+
)
|
|
4
|
+
from rapidata.rapidata_client.dataset.validation_set_builder import ValidationSetBuilder
|
|
2
5
|
from rapidata.rapidata_client.order.rapidata_order_builder import RapidataOrderBuilder
|
|
3
6
|
from rapidata.service.openapi_service import OpenAPIService
|
|
4
7
|
|
|
8
|
+
|
|
5
9
|
class RapidataClient:
|
|
6
10
|
"""
|
|
7
11
|
A client for interacting with the Rapidata API.
|
|
@@ -24,8 +28,6 @@ class RapidataClient:
|
|
|
24
28
|
client_id=client_id, client_secret=client_secret, endpoint=endpoint
|
|
25
29
|
)
|
|
26
30
|
|
|
27
|
-
|
|
28
|
-
|
|
29
31
|
def new_order(self, name: str) -> RapidataOrderBuilder:
|
|
30
32
|
"""
|
|
31
33
|
Create a new order using a RapidataOrderBuilder instance.
|
|
@@ -34,7 +36,6 @@ class RapidataClient:
|
|
|
34
36
|
:return: A RapidataOrderBuilder instance.
|
|
35
37
|
"""
|
|
36
38
|
return RapidataOrderBuilder(openapi_service=self.openapi_service, name=name)
|
|
37
|
-
|
|
38
39
|
|
|
39
40
|
def new_validation_set(self, name: str) -> ValidationSetBuilder:
|
|
40
41
|
"""
|
|
@@ -44,3 +45,12 @@ class RapidataClient:
|
|
|
44
45
|
:return: A ValidationDatasetBuilder instance.
|
|
45
46
|
"""
|
|
46
47
|
return ValidationSetBuilder(name=name, openapi_service=self.openapi_service)
|
|
48
|
+
|
|
49
|
+
def get_validation_set(self, validation_set_id: str) -> RapidataValidationSet:
|
|
50
|
+
"""
|
|
51
|
+
Get a validation set by ID.
|
|
52
|
+
|
|
53
|
+
:param validation_set_id: The ID of the validation set.
|
|
54
|
+
:return: The ValidationSet instance.
|
|
55
|
+
"""
|
|
56
|
+
return RapidataValidationSet(validation_set_id, self.openapi_service)
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
type RapidAsset = str | list[str]
|
|
@@ -193,24 +193,27 @@ rapidata/api_client/rest.py,sha256=A8BNcRsSo-Pn6sluvAOkTONtiA8gpulLjvGX93sQiAs,9
|
|
|
193
193
|
rapidata/rapidata_client/__init__.py,sha256=S34OtjhVlHBYmruPZHpJ5C-pVflCHPwj7FHwhMN-YyY,61
|
|
194
194
|
rapidata/rapidata_client/country_codes/__init__.py,sha256=Y8qeG2IMjvMGvhaPydq0nhwRQHb6dQqilctlEXu0_PE,55
|
|
195
195
|
rapidata/rapidata_client/country_codes/country_codes.py,sha256=Q0HMX7uHJQDeLCFPP5bq4iYi6pgcDWEcl2ONGhjgoeU,286
|
|
196
|
+
rapidata/rapidata_client/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
197
|
+
rapidata/rapidata_client/dataset/rapidata_dataset.py,sha256=mVx-YU3N6lvr_0NCg7yGodeBJh8A_Y0tFWRrxcFg9-w,2506
|
|
198
|
+
rapidata/rapidata_client/dataset/rapidata_validation_set.py,sha256=nfaENQOJZW5mZBY-4WAt-DFb3o46nN0J-QnHCLvEFPA,3308
|
|
199
|
+
rapidata/rapidata_client/dataset/validation_rapid_parts.py,sha256=96kb88XOwXuVI7stwY4Th4K3fo9-lFM-v7QV3qsS3NY,2073
|
|
200
|
+
rapidata/rapidata_client/dataset/validation_set_builder.py,sha256=UjKCgtQ23GM4CtiRrl1eQvD-eZFbiYCjdVf1iAVH8EI,5424
|
|
196
201
|
rapidata/rapidata_client/feature_flags/__init__.py,sha256=BNG_NQ4CrrC61fAWliImr8r581pIvegrkepVVbxcBw8,55
|
|
197
|
-
rapidata/rapidata_client/feature_flags/feature_flags.py,sha256=
|
|
202
|
+
rapidata/rapidata_client/feature_flags/feature_flags.py,sha256=uV4TPXc1oyqBRC9Ch3APprZwxBGnGMAl9DMFBUWmvnA,1048
|
|
198
203
|
rapidata/rapidata_client/metadata/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
199
204
|
rapidata/rapidata_client/metadata/base_metadata.py,sha256=BJoZT4lRs94lu7PE3__LcsMINHNDVKRKRART-EKzSaw,217
|
|
200
205
|
rapidata/rapidata_client/metadata/private_text_metadata.py,sha256=VJ_SJwp8OMFZLuUMRWvPeKOySXYJuGr-YgpwuLfTbmo,517
|
|
201
206
|
rapidata/rapidata_client/metadata/prompt_metadata.py,sha256=_FypjKWrC3iKUO_G2CVwAGcYbEYClVv4K7upqlnmecw,468
|
|
202
207
|
rapidata/rapidata_client/metadata/transcription_metadata.py,sha256=THtDEVCON4UlcXHmXrjilaOLHys4TrktUOPGWnXaCcc,631
|
|
203
208
|
rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
204
|
-
rapidata/rapidata_client/order/
|
|
205
|
-
rapidata/rapidata_client/order/
|
|
206
|
-
rapidata/rapidata_client/
|
|
207
|
-
rapidata/rapidata_client/order/rapidata_order.py,sha256=ve58PWrv396RxlpFo8OT3QeZvbJjLDRDkNxuCyZQkvg,3003
|
|
208
|
-
rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=eawFAmGKw8OlUlN1uZCxORAEJuepzsZk-Mzc82tu73o,6331
|
|
209
|
-
rapidata/rapidata_client/rapidata_client.py,sha256=cbn9d_kvOSLhgaGkUhQJ1lbjYfuw120Kw_x3Kyfzq4M,1607
|
|
209
|
+
rapidata/rapidata_client/order/rapidata_order.py,sha256=4a0Sn1y1UmL7HydhaDBOus-g_yILqPQMnTXfLEzQB5g,2997
|
|
210
|
+
rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=AnrBTwZoQpXpZ88qLQzdGEhqHy_XZ5-pFO-C_JKLdxo,6485
|
|
211
|
+
rapidata/rapidata_client/rapidata_client.py,sha256=BN3bfLEdNLYt7VZuEwDycNPT61wz66TTssjmNlRYlvU,2019
|
|
210
212
|
rapidata/rapidata_client/referee/__init__.py,sha256=x0AxGCsR6TlDjfqQ00lB9V7QVS9EZCJzweNEIzx42PI,207
|
|
211
213
|
rapidata/rapidata_client/referee/base_referee.py,sha256=bMy7cw0a-pGNbFu6u_1_Jplu0A483Ubj4oDQzh8vu8k,493
|
|
212
214
|
rapidata/rapidata_client/referee/classify_early_stopping_referee.py,sha256=-mPTCck81r2xn1GtMoizaOCftA3cOfj6YDSFbqsI3Ic,1140
|
|
213
215
|
rapidata/rapidata_client/referee/naive_referee.py,sha256=9CSoZou4YBFc9PiRvA2ELVQqyk-S1nug1SqhTwO9Tvw,719
|
|
216
|
+
rapidata/rapidata_client/types/__init__.py,sha256=Q9lzemtloA8e_XVVV1Y7ZZGsNg-xRqgj1GdJPrCzo5A,33
|
|
214
217
|
rapidata/rapidata_client/workflow/__init__.py,sha256=CDG8bKOBhLV4A0uBlqOd__79fH9KW3-UlZsR1RANLf0,250
|
|
215
218
|
rapidata/rapidata_client/workflow/base_workflow.py,sha256=335UyiNsIFix7ru_KZLeQHvfNg3GSId44ppPmWTXs_Y,1267
|
|
216
219
|
rapidata/rapidata_client/workflow/classify_workflow.py,sha256=m_0z3y3BNnoOeJv27GpmKXfyjVhjeK-5gaVbtYLboVc,1221
|
|
@@ -222,7 +225,7 @@ rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5
|
|
|
222
225
|
rapidata/service/openapi_service.py,sha256=xNlISI5NZByLy1Yeasc5Uafj1IA2mWJTzxUU1TtwyQU,1402
|
|
223
226
|
rapidata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
224
227
|
rapidata/utils/image_utils.py,sha256=TldO3eJWG8IhfJjm5MfNGO0mEDm1mQTsRoA0HLU1Uxs,404
|
|
225
|
-
rapidata-0.1.
|
|
226
|
-
rapidata-0.1.
|
|
227
|
-
rapidata-0.1.
|
|
228
|
-
rapidata-0.1.
|
|
228
|
+
rapidata-0.1.12.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
|
229
|
+
rapidata-0.1.12.dist-info/METADATA,sha256=DovtYrJwe4AJgFX-Zt5mMvfUiAMYcltIpMOCAc8ENpE,768
|
|
230
|
+
rapidata-0.1.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
|
|
231
|
+
rapidata-0.1.12.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|