mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-08-01 01:27:42 +00:00
Xu Yang
6e09dc6d69
* feat: add interactor classes and user interaction handling for experiments * update code * use fragment retry mechanism instead of rerun() * fix a bug * integrate user instructions into proposal and coder * fix CI * fix CI * feat: add approval option for user instructions submission * feat: enhance user instructions handling in Task and DSExperiment classes * fix CI * add user instructions into hypothesis rewrite * add interface to command line --------- Co-authored-by: Bowen Xian <xianbowen@outlook.com>
44 lines
1.8 KiB
Python
44 lines
1.8 KiB
Python
import re
|
|
from typing import Literal
|
|
|
|
import pandas as pd
|
|
|
|
from rdagent.core.experiment import Experiment, FBWorkspace, Task, UserInstructions
|
|
|
|
COMPONENT = Literal["DataLoadSpec", "FeatureEng", "Model", "Ensemble", "Workflow", "Pipeline"]
|
|
|
|
|
|
class DSExperiment(Experiment[Task, FBWorkspace, FBWorkspace]):
|
|
def __init__(self, pending_tasks_list: list, hypothesis_candidates: list | None = None, *args, **kwargs) -> None:
|
|
super().__init__(sub_tasks=[], *args, **kwargs)
|
|
# Status
|
|
# - Initial: blank;
|
|
# - Injecting from SOTA code;
|
|
# - New version no matter successful or not
|
|
# the initial workspace or the successful new version after coding
|
|
self.experiment_workspace = FBWorkspace()
|
|
self.pending_tasks_list = pending_tasks_list
|
|
self.hypothesis_candidates = hypothesis_candidates
|
|
|
|
self.format_check_result = None
|
|
# this field is optional. It is not none only when we have a format checker. Currently, only following cases are supported.
|
|
# - mle-bench
|
|
|
|
def set_user_instructions(self, user_instructions: UserInstructions | None):
|
|
super().set_user_instructions(user_instructions)
|
|
if user_instructions is None:
|
|
return
|
|
for task_list in self.pending_tasks_list:
|
|
for task in task_list:
|
|
task.user_instructions = user_instructions
|
|
|
|
def is_ready_to_run(self) -> bool:
|
|
"""
|
|
ready to run does not indicate the experiment is runnable
|
|
(so it is different from `trace.next_incomplete_component`.)
|
|
"""
|
|
return self.experiment_workspace is not None and "main.py" in self.experiment_workspace.file_dict
|
|
|
|
def set_local_selection(self, local_selection: tuple[int, ...]) -> None:
|
|
self.local_selection = local_selection
|