Skip to content
This repository was archived by the owner on Jul 29, 2024. It is now read-only.

Commit 1cd5606

Browse files
Update uploader.py
1 parent 54070c7 commit 1cd5606

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

labelpandas/uploader.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from labelbase.metadata import get_metadata_schema_to_name_key, process_metadata_value
2929
from labelbase.ontology import get_ontology_schema_to_name_path
3030
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
3232
from concurrent.futures import ThreadPoolExecutor, as_completed
3333

3434
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_
246246
top_level_name=annotation_index[column_name],
247247
annotation_inputs=row_dict[column_name],
248248
ontology_index=ontology_index,
249+
confidence=False,
249250
mask_method=mask_method,
250251
divider=divider
251252
)
@@ -256,10 +257,11 @@ def create_upload(row_dict:dict, row_data_col:str, global_key_col:str, external_
256257
ontology_index = model_run_id_to_ontology_index[modelRunId]
257258
for column_name in prediction_index.keys():
258259
predictions.extend(
259-
create_prediction_ndjsons(
260+
create_ndjsons(
260261
top_level_name=prediction_index[column_name],
261262
annotation_inputs=row_dict[column_name],
262263
ontology_index=ontology_index,
264+
confidence=True,
263265
mask_method=mask_method,
264266
divider=divider
265267
)

0 commit comments

Comments
 (0)