def new_task(self, task_id):
self.current_task_id = task_id
self.job_started_ts = timestamp_ms()
self.communication_ts = timestamp_ms()
return {
'current_task_id': self.current_task_id,
'job_started_ts': self.job_started_ts,
'communication_ts': self.communication_ts,
}
def __init__(self, name, platform_info, environment):
self.name = name
self.platform_info = platform_info
self.environment = environment
self.current_task_id = None
self.job_started_ts = None
self.error_logs = []
self.blocked_by_error = False
self.deleted = False
self.communication_ts = timestamp_ms()
self.register_ts = timestamp_ms()
def __init__(self, strategy_id, strategy_name, strategy_parameters,
shared_parameters):
self.strategy_id = strategy_id
self.strategy_name = strategy_name
self.strategy_parameters = strategy_parameters
self.shared_parameters = shared_parameters
self.created = timestamp_ms()
def __init__(self, experiment_id, assemble_task_id, learn_task_id):
self.experiment_id = experiment_id
self.assemble_task_id = assemble_task_id
self.learn_task_id = learn_task_id
self.created_ts = timestamp_ms()
# state is "generated", "running" or "completed"
self.state = "generated"
self.completed_ts = None
def completed(self, task_id):
if task_id == self.learn_task_id or (task_id == self.assemble_task_id and not self.learn_task_id):
self.completed_ts = timestamp_ms()
self.state = "completed"
return {
'completed_ts': self.completed_ts,
'state': self.state,
}
def __init__(self, experiment_id, assemble_task_id, learn_task_id, order_in_exp=0):
self.experiment_id = experiment_id
self.assemble_task_id = assemble_task_id
self.learn_task_id = learn_task_id
self.order_in_exp = order_in_exp
self.created_ts = timestamp_ms()
# state values: "generated", "runnable", "running", "completed"
self.state = "generated"
self.completed_ts = None
def clear_task(self):
self.current_task_id = None
self.job_started_ts = None
self.communication_ts = timestamp_ms()
return {
'current_task_id': self.current_task_id,
'job_started_ts': self.job_started_ts,
'communication_ts': self.communication_ts,
}
def __init__(self, name, markdown):
self.name = name
self.markdown = markdown
# state is "configuration", "generating_tasks" or "generated_tasks"
self.state = "configuration"
self.assemble_configs = []
self.learn_configs = []
self.tasks = []
self.created_ts = timestamp_ms()
def add_log(self, progress, message):
self.progress = progress
self.log.append({
"timestamp": timestamp_ms(),
"message": message,
})
return {
'progress': self.progress,
'log': self.log,
}
def completed(self, result):
self.result = result
self.progress = 1.
self.completed_ts = timestamp_ms()
self.state = "completed"
return {
'result': self.result,
'progress': self.progress,
'completed_ts': self.completed_ts,
'state': self.state,
}
def error(self, log):
self.blocked_by_error = True
result = {'blocked_by_error': self.blocked_by_error}
if log:
self.error_logs.append({
"timestamp": timestamp_ms(),
"message": log,
})
result.update({'error_logs': self.error_logs})
return result
def __init__(self, assemble_config):
self.assemble_config = assemble_config
# state is "generated", "runnable", "running" or "completed"
self.state = "generated"
self.assigned_to = None
self.result_file_path = None
self.created_ts = timestamp_ms()
self.completed_ts = None
self.progress = None
self.log = None
self.up_to_date = True
self.parent_tasks = []
def __init__(self, assemble_config, order_in_exp=0):
self.assemble_config = assemble_config
self.order_in_exp = order_in_exp
# state values: "generated", "runnable", "running", "completed"
self.state = "generated"
self.assigned_to = None
self.result_file_path = None
self.created_ts = timestamp_ms()
self.completed_ts = None
self.progress = None
self.log = None
self.up_to_date = True
def completed(self, result):
if result:
self.result_file_path = result
self.progress = 1.
self.completed_ts = timestamp_ms()
self.state = "completed"
return {
'result_file_path': self.result_file_path,
'progress': self.progress,
'completed_ts': self.completed_ts,
'state': self.state,
}
def re_run(self, assemble_task_id, learn_task_id):
self.assemble_task_id = assemble_task_id
self.learn_task_id = learn_task_id
self.created_ts = timestamp_ms()
self.state = "running"
self.completed_ts = None
return {
'assemble_task_id': self.assemble_task_id,
'learn_task_id': self.learn_task_id,
'created_ts': self.created_ts,
'state': self.state,
'completed_ts': self.completed_ts,
}
def ping(self):
self.communication_ts = timestamp_ms()
return {
'communication_ts': self.communication_ts,
}