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

Commit 1655698

Browse files
Update client.py
1 parent d1f1d30 commit 1655698

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

labelpandas/client.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from labelbox import Client as labelboxClient
2-
from labelbase import Client as labelbaseClient
32
from labelbox.schema.dataset import Dataset as labelboxDataset
3+
from labelbase.metadata import sync_metadata_fields
4+
from labelbase.uploaders import batch_create_data_rows
45
import pandas as pd
56
from labelpandas import connector
67

@@ -22,7 +23,6 @@ def __init__(
2223
lb_app_url="https://app.labelbox.com"):
2324

2425
self.lb_client = labelboxClient(lb_api_key, endpoint=lb_endpoint, enable_experimental=lb_enable_experimental, app_url=lb_app_url)
25-
self.base_client = labelbaseClient(lb_api_key, lb_endpoint=lb_endpoint, lb_enable_experimental=lb_enable_experimental, lb_app_url=lb_app_url)
2626

2727
# def create_table_from_dataset():
2828
# return table
@@ -54,8 +54,8 @@ def create_data_rows_from_table(
5454
"""
5555

5656
# Ensure all your metadata_index keys are metadata fields in Labelbox and that your Pandas DataFrame has all the right columns
57-
table = self.base_client.sync_metadata_fields(
58-
table=table, get_columns_function=connector.get_columns_function, add_column_function=connector.add_column_function,
57+
table = sync_metadata_fields(
58+
client=self.lb_client, table=table, get_columns_function=connector.get_columns_function, add_column_function=connector.add_column_function,
5959
get_unique_values_function=connector.get_unique_values_function, metadata_index=metadata_index, verbose=verbose
6060
)
6161

@@ -65,7 +65,7 @@ def create_data_rows_from_table(
6565

6666
# Create a dictionary where {key=global_key : value=labelbox_upload_dictionary} - this is unique to Pandas
6767
global_key_to_upload_dict, conversion_errors = connector.create_upload_dict(
68-
table=table, lb_client=self.lb_client, base_client=self.base_client,
68+
table=table, lb_client=self.lb_client,
6969
row_data_col=row_data_col, global_key_col=global_key_col, external_id_col=external_id_col,
7070
metadata_index=metadata_index, local_files=local_files, divider=divider, verbose=verbose
7171
)
@@ -80,8 +80,8 @@ def create_data_rows_from_table(
8080
return {"upload_results" : [], "conversion_errors" : errors}
8181

8282
# Upload your data rows to Labelbox
83-
upload_results = self.base_client.batch_create_data_rows(
84-
dataset=lb_dataset, global_key_to_upload_dict=global_key_to_upload_dict,
83+
upload_results = batch_create_data_rows(
84+
client=self.lb_client, dataset=lb_dataset, global_key_to_upload_dict=global_key_to_upload_dict,
8585
skip_duplicates=skip_duplicates, divider=divider, verbose=verbose
8686
)
8787

0 commit comments

Comments
 (0)