28
28
from labelbase .metadata import get_metadata_schema_to_name_key , process_metadata_value
29
29
from labelbase .ontology import get_ontology_schema_to_name_path
30
30
from labelbase .models import create_model_run_with_name
31
- from labelbase .annotate import create_ndjsons , create_prediction_ndjsons
31
+ from labelbase .annotate import create_ndjsons
32
32
from concurrent .futures import ThreadPoolExecutor , as_completed
33
33
34
34
def create_upload_dict (client :labelboxClient , table : pandas .core .frame .DataFrame , table_dict :dict ,
@@ -246,6 +246,7 @@ def create_upload(row_dict:dict, row_data_col:str, global_key_col:str, external_
246
246
top_level_name = annotation_index [column_name ],
247
247
annotation_inputs = row_dict [column_name ],
248
248
ontology_index = ontology_index ,
249
+ confidence = False ,
249
250
mask_method = mask_method ,
250
251
divider = divider
251
252
)
@@ -256,10 +257,11 @@ def create_upload(row_dict:dict, row_data_col:str, global_key_col:str, external_
256
257
ontology_index = model_run_id_to_ontology_index [modelRunId ]
257
258
for column_name in prediction_index .keys ():
258
259
predictions .extend (
259
- create_prediction_ndjsons (
260
+ create_ndjsons (
260
261
top_level_name = prediction_index [column_name ],
261
262
annotation_inputs = row_dict [column_name ],
262
263
ontology_index = ontology_index ,
264
+ confidence = True ,
263
265
mask_method = mask_method ,
264
266
divider = divider
265
267
)
0 commit comments