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

Commit 60072d5

Browse files
Update client.py
1 parent a7b49ae commit 60072d5

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

labelpandas/client.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,11 @@ def __init__(
2828
# return table
2929

3030
def create_data_rows_from_table(
31-
self, df:pd.core.frame.DataFrame, lb_dataset:labelboxDataset, row_data_col:str, global_key_col=None, external_id_col=None,
31+
self, dataframe:pd.core.frame.DataFrame, lb_dataset:labelboxDataset, row_data_col:str, global_key_col=None, external_id_col=None,
3232
metadata_index:dict={}, local_files:bool=False, skip_duplicates:bool=False, verbose:bool=False, divider="___"):
3333
""" Creates Labelbox data rows given a Pandas table and a Labelbox Dataset
3434
Args:
35-
df : Required (pandas.core.frame.DataFrame) - Pandas DataFrame
35+
dataframe : Required (pandas.core.frame.DataFrame) - Pandas DataFrame
3636
lb_dataset : Required (labelbox.schema.dataset.Dataset) - Labelbox dataset to add data rows to
3737
row_data_col : Required (str) - Column containing asset URL or file path
3838
local_files : Required (bool) - Determines how to handle row_data_col values
@@ -54,18 +54,18 @@ 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-
df = self.base_client.sync_metadata_fields(
58-
table=df, get_columns_function=connector.get_columns_function, add_column_function=connector.add_column_function,
57+
dataframe = self.base_client.sync_metadata_fields(
58+
table=dataframe, 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

6262
# If df returns False, the sync failed - terminate the upload
63-
if type(df) == bool:
63+
if type(dataframe) == bool:
6464
return {"upload_results" : [], "conversion_errors" : []}
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-
df=df, lb_client=self.lb_client, base_client=self.base_client,
68+
table=dataframe, lb_client=self.lb_client, base_client=self.base_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
)

0 commit comments

Comments
 (0)