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

Commit 3218c3e

Browse files
Update connector.py
1 parent 79da1a2 commit 3218c3e

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

labelpandas/connector.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from labelbase import Client as baseClient
33
import pandas
44
from concurrent.futures import ThreadPoolExecutor, as_completed
5-
import math
5+
import numpy
66

77
def create_upload_dict(df:pandas.core.frame.DataFrame, local_files:bool, lb_client:Client, base_client:baseClient, row_data_col:str,
88
global_key_col:str="", external_id_col:str="", metadata_index:dict={}, divider:str="///", verbose=False):
@@ -72,10 +72,10 @@ def create_data_rows(local_files:bool, lb_client:Client, row:pandas.core.series.
7272
for metadata_field_name in metadata_index.keys():
7373
row_value = row[metadata_field_name]
7474
metadata_type = metadata_index[metadata_field_name]
75-
if math.isnan(row_value):
76-
continue
77-
elif row_value:
78-
if metadata_type == "enum":
75+
if row_value:
76+
if str(row_value) == "nan":
77+
continue
78+
elif metadata_type == "enum":
7979
name_key = f"{metadata_field_name}{divider}{row[metadata_field_name]}"
8080
value = metadata_name_key_to_schema[name_key]
8181
elif metadata_type == "number":

0 commit comments

Comments
 (0)