File tree Expand file tree Collapse file tree 4 files changed +6
-4
lines changed Expand file tree Collapse file tree 4 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 10
10
ANNOTATION_TYPES = (BOX_TYPE , POLYGON_TYPE , SEGMENTATION_TYPE )
11
11
ANNOTATION_UPDATE_KEY = "update"
12
12
AUTOTAGS_KEY = "autotags"
13
-
13
+ EXPORTED_ROWS = "exportedRows"
14
14
CLASS_PDF_KEY = "class_pdf"
15
15
CONFIDENCE_KEY = "confidence"
16
16
DATASET_ID_KEY = "dataset_id"
Original file line number Diff line number Diff line change 17
17
DATASET_NAME_KEY ,
18
18
DATASET_SLICES_KEY ,
19
19
DEFAULT_ANNOTATION_UPDATE_MODE ,
20
+ EXPORTED_ROWS ,
20
21
JOB_ID_KEY ,
21
22
NAME_KEY ,
22
23
REFERENCE_IDS_KEY ,
@@ -347,4 +348,4 @@ def items_and_annotations(
347
348
route = f"dataset/{ self .id } /exportForTraining" ,
348
349
requests_command = requests .get ,
349
350
)
350
- return convert_export_payload (api_payload ["exportedRows" ])
351
+ return convert_export_payload (api_payload [EXPORTED_ROWS ])
Original file line number Diff line number Diff line change 6
6
from nucleus .dataset_item import DatasetItem
7
7
from nucleus .job import AsyncJob
8
8
from nucleus .utils import convert_export_payload , format_dataset_item_response
9
+ from nucleus .constants import EXPORTED_ROWS
9
10
10
11
11
12
class Slice :
@@ -115,7 +116,7 @@ def items_and_annotations(
115
116
route = f"slice/{ self .slice_id } /exportForTraining" ,
116
117
requests_command = requests .get ,
117
118
)
118
- return convert_export_payload (api_payload ["exportedRows" ])
119
+ return convert_export_payload (api_payload [EXPORTED_ROWS ])
119
120
120
121
def send_to_labeling (self , project_id : str ):
121
122
response = self ._client .make_request (
Original file line number Diff line number Diff line change @@ -117,7 +117,7 @@ def reference_id_from_url(url):
117
117
{
118
118
** TEST_POLYGON_ANNOTATIONS [i ],
119
119
"confidence" : 0.10 * i ,
120
- "class_pdf" : None ,
120
+ "class_pdf" : TEST_POLYGON_MODEL_PDF ,
121
121
}
122
122
if i != 0
123
123
else {
You can’t perform that action at this time.
0 commit comments