Commit 31b2b886 authored by liuzhe-lz's avatar liuzhe-lz Committed by chicm-ms
Browse files

Fix bug introduced in customized trial (#1779)

parent 48084d88
......@@ -184,7 +184,7 @@ class MsgDispatcher(MsgDispatcherBase):
"""
id_ = data['parameter_id']
value = data['value']
if not id_ or id_ in _customized_parameter_ids:
if id_ is None or id_ in _customized_parameter_ids:
if not hasattr(self.tuner, '_accept_customized'):
self.tuner._accept_customized = False
if not self.tuner._accept_customized:
......
import logging
import time
from nni.tuner import Tuner
......@@ -7,14 +8,18 @@ class MultiThreadTuner(Tuner):
self.parent_done = False
def generate_parameters(self, parameter_id, **kwargs):
logging.debug('generate_parameters: %s %s', parameter_id, kwargs)
if parameter_id == 0:
return {'x': 0}
else:
while not self.parent_done:
logging.debug('parameter_id %s sleeping', parameter_id)
time.sleep(2)
logging.debug('parameter_id %s waked up', parameter_id)
return {'x': 1}
def receive_trial_result(self, parameter_id, parameters, value, **kwargs):
logging.debug('receive_trial_result: %s %s %s %s', parameter_id, parameters, value, kwargs)
if parameter_id == 0:
self.parent_done = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment