fix bug in NaN support (#2077)

src/sdk/pynni/nni/bohb_advisor/bohb_advisor.py

@@ -557,7 +557,8 @@ class BOHB(MsgDispatcherBase):
             Data type not supported
         """
         logger.debug('handle report metric data = %s', data)
-
+        if 'value' in data:
+            data['value'] = json_tricks.loads(data['value'])
         if data['type'] == MetricType.REQUEST_PARAMETER:
             assert multi_phase_enabled()
             assert data['trial_job_id'] is not None
@@ -627,6 +628,8 @@ class BOHB(MsgDispatcherBase):
         AssertionError
             data doesn't have required key 'parameter' and 'value'
         """
+        for entry in data:
+            entry['value'] = json_tricks.loads(entry['value'])
         _completed_num = 0
         for trial_info in data:
             logger.info("Importing data, current processing progress %s / %s", _completed_num, len(data))

src/sdk/pynni/nni/hyperband_advisor/hyperband_advisor.py

@@ -380,6 +380,8 @@ class Hyperband(MsgDispatcherBase):
         ValueError
             Data type not supported
         """
+        if 'value' in data:
+            data['value'] = json_tricks.loads(data['value'])
         if data['type'] == MetricType.REQUEST_PARAMETER:
             assert multi_phase_enabled()
             assert data['trial_job_id'] is not None

src/sdk/pynni/nni/msg_dispatcher.py

@@ -113,6 +113,8 @@ class MsgDispatcher(MsgDispatcherBase):
         """Import additional data for tuning
         data: a list of dictionaries, each of which has at least two keys, 'parameter' and 'value'
         """
+        for entry in data:
+            entry['value'] = json_tricks.loads(entry['value'])
         self.tuner.import_data(data)
 
     def handle_add_customized_trial(self, data):
@@ -128,7 +130,8 @@ class MsgDispatcher(MsgDispatcherBase):
               - 'type': report type, support {'FINAL', 'PERIODICAL'}
         """
         # metrics value is dumped as json string in trial, so we need to decode it here
-        data['value'] = json_tricks.loads(data['value'])
+        if 'value' in data:
+            data['value'] = json_tricks.loads(data['value'])
         if data['type'] == MetricType.FINAL:
             self._handle_final_metric_data(data)
         elif data['type'] == MetricType.PERIODICAL: