Skip to content

Commit 5e0a52f

Browse files
committed
Fix for [sc-222934]
1 parent 53ce60d commit 5e0a52f

2 files changed

Lines changed: 14 additions & 2 deletions

File tree

python-lib/dku_utils.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import json
22
import copy
3+
import math
34
from jsonpath_ng.ext import parse
45
from safe_logger import SafeLogger
56

@@ -154,3 +155,10 @@ def decode_csv_data(data):
154155
logger.error("Could not extract csv data. Error={}".format(error))
155156
json_data = data
156157
return json_data
158+
159+
160+
def de_NaN(input):
161+
if isinstance(input, float):
162+
if math.isnan(input):
163+
return ''
164+
return input

python-lib/rest_api_recipe_session.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from dataikuapi.utils import DataikuException
22
from rest_api_client import RestAPIClient
33
from safe_logger import SafeLogger
4-
from dku_utils import parse_keys_for_json, get_value_from_path, decode_csv_data
4+
from dku_utils import parse_keys_for_json, get_value_from_path, decode_csv_data, de_NaN
55
from dku_constants import DKUConstants
66
import copy
77
import json
@@ -49,7 +49,11 @@ def process_dataframe(self, input_parameters_dataframe, is_raw_output):
4949
self.initial_parameter_columns = {}
5050
for column_name in self.column_to_parameter_dict:
5151
parameter_name = self.column_to_parameter_dict[column_name]
52-
self.initial_parameter_columns.update({parameter_name: input_parameters_row.get(column_name)})
52+
self.initial_parameter_columns.update(
53+
{
54+
parameter_name: de_NaN(input_parameters_row.get(column_name))
55+
}
56+
)
5357
updated_endpoint_parameters = copy.deepcopy(self.endpoint_parameters)
5458
updated_endpoint_parameters.update(self.initial_parameter_columns)
5559
logger.info("Processing row #{}, creating client with credential={}, updated_endpoint={}, custom_key_values={}".format(

0 commit comments

Comments
 (0)