SYMBOL INDEX (3037 symbols across 506 files)

FILE: rdagent/app/CI/run.py
  class CIError (line 43) | class CIError:
    method to_dict (line 53) | def to_dict(self) -> dict[str, object]:
    method __str__ (line 56) | def __str__(self) -> str:
  class CIFeedback (line 61) | class CIFeedback(Feedback):
    method statistics (line 64) | def statistics(self) -> dict[Literal["ruff", "mypy"], dict[str, int]]:
  class FixRecord (line 73) | class FixRecord:
    method to_dict (line 79) | def to_dict(self) -> dict[str, Any]:
  class CodeFile (line 90) | class CodeFile:
    method __init__ (line 91) | def __init__(self, path: Path | str) -> None:
    method add_line_number (line 96) | def add_line_number(cls: CodeFile, code: list[str] | str, start: int =...
    method remove_line_number (line 107) | def remove_line_number(cls: CodeFile, code: list[str] | str) -> list[s...
    method load (line 117) | def load(self) -> None:
    method get (line 126) | def get(
    method apply_changes (line 159) | def apply_changes(self, changes: list[tuple[int, int, str]]) -> None:
    method get_code_blocks (line 182) | def get_code_blocks(self, max_lines: int = 30) -> list[tuple[int, int]]:
    method __str__ (line 214) | def __str__(self) -> str:
  class Repo (line 218) | class Repo(EvolvableSubjects):
    method __init__ (line 219) | def __init__(self, project_path: Path | str, excludes: list[Path] | No...
  class RuffRule (line 255) | class RuffRule:
  class RuffEvaluator (line 282) | class RuffEvaluator(Evaluator):
    method __init__ (line 287) | def __init__(self, command: str | None = None) -> None:
    method explain_rule (line 294) | def explain_rule(error_code: str) -> RuffRule:
    method evaluate (line 307) | def evaluate(self, evo: Repo, **kwargs: dict) -> CIFeedback:
  class MypyEvaluator (line 358) | class MypyEvaluator(Evaluator):
    method __init__ (line 359) | def __init__(self, command: str | None = None) -> None:
    method evaluate (line 365) | def evaluate(self, evo: Repo, **kwargs: dict) -> CIFeedback:
  class MultiEvaluator (line 413) | class MultiEvaluator(Evaluator):
    method __init__ (line 414) | def __init__(self, *evaluators: Evaluator) -> None:
    method evaluate (line 417) | def evaluate(self, evo: Repo, **kwargs: dict) -> CIFeedback:
  class CIEvoStr (line 431) | class CIEvoStr(EvolvingStrategy):
    method evolve (line 432) | def evolve(  # noqa: C901, PLR0912, PLR0915
  class CIEvoAgent (line 696) | class CIEvoAgent(EvoAgent):
    method __init__ (line 697) | def __init__(self, evolving_strategy: CIEvoStr) -> None:
    method multistep_evolve (line 701) | def multistep_evolve(self, evo: Repo, eva: Evaluator) -> Repo:

FILE: rdagent/app/benchmark/factor/analysis.py
  class BenchmarkAnalyzer (line 15) | class BenchmarkAnalyzer:
    method __init__ (line 16) | def __init__(self, settings, only_correct_format=False):
    method load_index_map (line 21) | def load_index_map(self):
    method load_data (line 29) | def load_data(self, file_path):
    method process_results (line 39) | def process_results(self, results):
    method reformat_index (line 48) | def reformat_index(self, display_df):
    method result_all_key_order (line 81) | def result_all_key_order(self, x):
    method analyze_data (line 96) | def analyze_data(self, sum_df):
  class Plotter (line 178) | class Plotter:
    method change_fs (line 180) | def change_fs(font_size):
    method plot_data (line 190) | def plot_data(data, file_name, title):
  function main (line 204) | def main(

FILE: rdagent/app/cli.py
  function ui (line 45) | def ui(port=19899, log_dir="", debug: bool = False, data_science: bool =...
  function server_ui (line 65) | def server_ui(port=19899):
  function ds_user_interact (line 74) | def ds_user_interact(port=19900):
  function fin_factor_cli (line 83) | def fin_factor_cli(
  function fin_model_cli (line 94) | def fin_model_cli(
  function fin_quant_cli (line 105) | def fin_quant_cli(
  function fin_factor_report_cli (line 116) | def fin_factor_report_cli(
  function general_model_cli (line 126) | def general_model_cli(report_file_path: str):
  function data_science_cli (line 131) | def data_science_cli(
  function llm_finetune_cli (line 150) | def llm_finetune_cli(
  function grade_summary_cli (line 177) | def grade_summary_cli(log_folder: str):
  function health_check_cli (line 186) | def health_check_cli(
  function collect_info_cli (line 195) | def collect_info_cli():

FILE: rdagent/app/data_science/conf.py
  class DataScienceBasePropSetting (line 9) | class DataScienceBasePropSetting(KaggleBasePropSetting):

FILE: rdagent/app/data_science/loop.py
  function main (line 13) | def main(

FILE: rdagent/app/finetune/data_science/conf.py
  class DSFinetuneScen (line 9) | class DSFinetuneScen(ExtendedBaseSettings):
  function update_settings (line 30) | def update_settings(competition: str):

FILE: rdagent/app/finetune/data_science/loop.py
  function main (line 13) | def main(

FILE: rdagent/app/finetune/data_science/scen.py
  class DSFinetuneScen (line 11) | class DSFinetuneScen(DataScienceScen):
    method _get_data_folder_description (line 14) | def _get_data_folder_description(self) -> str:

FILE: rdagent/app/finetune/llm/conf.py
  class LLMFinetunePropSetting (line 8) | class LLMFinetunePropSetting(ExtendedBaseSettings):

FILE: rdagent/app/finetune/llm/loop.py
  function main (line 17) | def main(

FILE: rdagent/app/finetune/llm/ui/app.py
  function get_job_options (line 28) | def get_job_options(base_path: Path) -> list[str]:
  function main (line 66) | def main():

FILE: rdagent/app/finetune/llm/ui/benchmarks/__init__.py
  function get_core_metric_score (line 22) | def get_core_metric_score(benchmark_name: str, accuracy_summary: dict) -...

FILE: rdagent/app/finetune/llm/ui/benchmarks/base.py
  class BenchmarkProcessor (line 6) | class BenchmarkProcessor(ABC):
    method match (line 28) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 34) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...
    method is_higher_better (line 49) | def is_higher_better(cls, metric_name: str) -> bool:

FILE: rdagent/app/finetune/llm/ui/benchmarks/bioprobench.py
  class BioProBenchProcessor (line 6) | class BioProBenchProcessor(BenchmarkProcessor):
    method match (line 17) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 21) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...

FILE: rdagent/app/finetune/llm/ui/benchmarks/chemcotbench.py
  class ChemCotBenchProcessor (line 6) | class ChemCotBenchProcessor(BenchmarkProcessor):
    method match (line 43) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 47) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...

FILE: rdagent/app/finetune/llm/ui/benchmarks/financeiq.py
  class FinanceIQProcessor (line 6) | class FinanceIQProcessor(BenchmarkProcessor):
    method match (line 10) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 14) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...

FILE: rdagent/app/finetune/llm/ui/benchmarks/panorama.py
  class PanoramaProcessor (line 6) | class PanoramaProcessor(BenchmarkProcessor):
    method match (line 16) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 20) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...

FILE: rdagent/app/finetune/llm/ui/benchmarks/tablebench.py
  class TableBenchProcessor (line 6) | class TableBenchProcessor(BenchmarkProcessor):
    method match (line 24) | def match(cls, benchmark_name: str) -> bool:
    method get_core_metric (line 28) | def get_core_metric(cls, accuracy_summary: dict) -> tuple[str, float, ...

FILE: rdagent/app/finetune/llm/ui/components.py
  function convert_latex_for_streamlit (line 17) | def convert_latex_for_streamlit(text: str) -> str:
  function format_duration (line 32) | def format_duration(seconds: float | None) -> str:
  function render_session (line 42) | def render_session(session: Session, show_types: list[str]) -> None:
  function render_loop (line 58) | def render_loop(loop: Loop, show_types: list[str]) -> None:
  function _render_loop_content (line 124) | def _render_loop_content(loop: Loop, show_types: list[str]) -> None:
  function render_evo_loop (line 158) | def render_evo_loop(evo: EvoLoop, show_types: list[str]) -> None:
  function render_event (line 170) | def render_event(event: Event) -> None:
  function render_scenario (line 208) | def render_scenario(content: Any) -> None:
  function render_dataset_selection (line 279) | def render_dataset_selection(content: Any) -> None:
  function render_hypothesis (line 296) | def render_hypothesis(content: Any) -> None:
  function render_settings (line 306) | def render_settings(content: Any) -> None:
  function render_llm_call (line 313) | def render_llm_call(content: Any) -> None:
  function render_template (line 354) | def render_template(content: Any) -> None:
  function render_experiment (line 378) | def render_experiment(content: Any) -> None:
  function render_code (line 392) | def render_code(content: Any) -> None:
  function _extract_evaluator_name (line 410) | def _extract_evaluator_name(title: str) -> str:
  function _render_single_feedback (line 416) | def _render_single_feedback(fb: Any, evaluator_name: str = "") -> None:
  function render_docker_exec (line 461) | def render_docker_exec(content: Any, event_title: str = "") -> None:
  function render_feedback (line 530) | def render_feedback(content: Any) -> None:
  function render_token (line 572) | def render_token(content: Any) -> None:
  function render_time_info (line 585) | def render_time_info(content: Any) -> None:
  function render_generic (line 593) | def render_generic(content: Any) -> None:
  function render_training_result (line 602) | def render_training_result(result: dict) -> None:
  function render_benchmark_result (line 665) | def render_benchmark_result(content: dict) -> None:
  function render_summary (line 709) | def render_summary(summary: dict) -> None:

FILE: rdagent/app/finetune/llm/ui/data_loader.py
  class Event (line 19) | class Event:
    method time_str (line 34) | def time_str(self) -> str:
  class EvoLoop (line 39) | class EvoLoop:
  class Loop (line 48) | class Loop:
  class Session (line 59) | class Session:
  function extract_loop_id (line 66) | def extract_loop_id(tag: str) -> int | None:
  function extract_evo_id (line 71) | def extract_evo_id(tag: str) -> int | None:
  function extract_stage (line 76) | def extract_stage(tag: str) -> str:
  function get_valid_sessions (line 88) | def get_valid_sessions(log_folder: Path) -> list[str]:
  function parse_event (line 98) | def parse_event(tag: str, content: Any, timestamp: datetime) -> Event | ...
  function load_ft_session (line 365) | def load_ft_session(log_path: Path) -> Session:
  function get_summary (line 423) | def get_summary(session: Session) -> dict:

FILE: rdagent/app/finetune/llm/ui/ft_summary.py
  function is_valid_task (line 16) | def is_valid_task(task_path: Path) -> bool:
  function get_loop_dirs (line 21) | def get_loop_dirs(task_path: Path) -> list[Path]:
  function extract_benchmark_score (line 27) | def extract_benchmark_score(loop_path: Path, split: str = "") -> tuple[s...
  function extract_benchmark_scores (line 60) | def extract_benchmark_scores(loop_path: Path) -> dict[str, tuple[str, fl...
  function extract_baseline_score (line 73) | def extract_baseline_score(task_path: Path) -> tuple[str, float] | None:
  function extract_baseline_scores (line 101) | def extract_baseline_scores(task_path: Path) -> dict[str, tuple[str, flo...
  function get_loop_status (line 144) | def get_loop_status(
  function get_max_loops (line 214) | def get_max_loops(job_path: Path) -> int:
  function get_job_summary_df (line 224) | def get_job_summary_df(job_path: Path) -> tuple[pd.DataFrame, pd.DataFra...
  function style_status_cell (line 318) | def style_status_cell(val: str, decision: bool | None = None) -> str:
  function style_df_with_decisions (line 364) | def style_df_with_decisions(df: pd.DataFrame, decisions_df: pd.DataFrame...
  function render_job_summary (line 386) | def render_job_summary(job_path: Path, is_root: bool = False) -> None:
  function extract_full_benchmark (line 429) | def extract_full_benchmark(loop_path: Path, split: str = "") -> dict | N...
  function extract_baseline_full_benchmark (line 457) | def extract_baseline_full_benchmark(task_path: Path, split: str = "valid...
  function get_task_full_benchmark_df (line 490) | def get_task_full_benchmark_df(task_path: Path, split: str) -> pd.DataFr...
  function render_task_detail_selector (line 554) | def render_task_detail_selector(job_path: Path) -> None:

FILE: rdagent/app/finetune/share/eval.py
  class PrevModelLoadEvaluator (line 13) | class PrevModelLoadEvaluator(CoSTEEREvaluator):
    method __init__ (line 16) | def __init__(self, scen: Scenario):
    method evaluate (line 19) | def evaluate(

FILE: rdagent/app/general_model/general_model.py
  function extract_models_and_implement (line 14) | def extract_models_and_implement(report_file_path: str) -> None:

FILE: rdagent/app/kaggle/conf.py
  class KaggleBasePropSetting (line 6) | class KaggleBasePropSetting(ExtendedBaseSettings):

FILE: rdagent/app/kaggle/loop.py
  class KaggleRDLoop (line 29) | class KaggleRDLoop(RDLoop):
    method __init__ (line 30) | def __init__(self, PROP_SETTING: BasePropSetting):
    method coding (line 58) | def coding(self, prev_out: dict[str, Any]):
    method running (line 71) | def running(self, prev_out: dict[str, Any]):
  function main (line 114) | def main(path=None, step_n=None, competition=None):

FILE: rdagent/app/qlib_rd_loop/conf.py
  class ModelBasePropSetting (line 8) | class ModelBasePropSetting(BasePropSetting):
  class FactorBasePropSetting (line 52) | class FactorBasePropSetting(BasePropSetting):
  class FactorFromReportPropSetting (line 96) | class FactorFromReportPropSetting(FactorBasePropSetting):
  class QuantBasePropSetting (line 112) | class QuantBasePropSetting(BasePropSetting):

FILE: rdagent/app/qlib_rd_loop/factor.py
  class FactorRDLoop (line 17) | class FactorRDLoop(RDLoop):
    method running (line 21) | def running(self, prev_out: dict[str, Any]):
  function main (line 30) | def main(

FILE: rdagent/app/qlib_rd_loop/factor_from_report.py
  function generate_hypothesis (line 26) | def generate_hypothesis(factor_result: dict, report_content: str) -> str:
  function extract_hypothesis_and_exp_from_reports (line 61) | def extract_hypothesis_and_exp_from_reports(report_file_path: str) -> Ql...
  class FactorReportLoop (line 97) | class FactorReportLoop(FactorRDLoop, metaclass=LoopMeta):
    method __init__ (line 98) | def __init__(self, report_folder: str = None):
    method direct_exp_gen (line 112) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 137) | def coding(self, prev_out: dict[str, Any]):
  function main (line 143) | def main(report_folder=None, path=None, all_duration=None, checkout=True):

FILE: rdagent/app/qlib_rd_loop/model.py
  class ModelRDLoop (line 14) | class ModelRDLoop(RDLoop):
  function main (line 18) | def main(

FILE: rdagent/app/qlib_rd_loop/quant.py
  class QuantRDLoop (line 30) | class QuantRDLoop(RDLoop):
    method __init__ (line 36) | def __init__(self, PROP_SETTING: BasePropSetting):
    method direct_exp_gen (line 74) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 92) | def coding(self, prev_out: dict[str, Any]):
    method running (line 100) | def running(self, prev_out: dict[str, Any]):
    method feedback (line 111) | def feedback(self, prev_out: dict[str, Any]):
  function main (line 131) | def main(

FILE: rdagent/app/rl/conf.py
  class RLPostTrainingPropSetting (line 8) | class RLPostTrainingPropSetting(ExtendedBaseSettings):

FILE: rdagent/app/rl/loop.py
  function main (line 16) | def main(

FILE: rdagent/app/rl/ui/app.py
  function _safe_resolve (line 23) | def _safe_resolve(user_input: str | None, safe_root: Path) -> Path:
  function get_job_options (line 46) | def get_job_options(base_path: Path) -> list[str]:
  function main (line 75) | def main():

FILE: rdagent/app/rl/ui/components.py
  function format_duration (line 14) | def format_duration(seconds: float | None) -> str:
  function render_session (line 24) | def render_session(session: Session, show_types: list[str]) -> None:
  function render_loop (line 38) | def render_loop(loop: Loop, show_types: list[str]) -> None:
  function render_event (line 95) | def render_event(event: Event) -> None:
  function render_scenario (line 127) | def render_scenario(content: Any) -> None:
  function render_hypothesis (line 135) | def render_hypothesis(content: Any) -> None:
  function render_settings (line 144) | def render_settings(content: Any) -> None:
  function render_llm_call (line 151) | def render_llm_call(content: Any) -> None:
  function render_template (line 181) | def render_template(content: Any) -> None:
  function render_experiment (line 200) | def render_experiment(content: Any) -> None:
  function render_code (line 211) | def render_code(content: Any) -> None:
  function render_docker_exec (line 228) | def render_docker_exec(content: Any) -> None:
  function render_feedback (line 245) | def render_feedback(content: Any) -> None:
  function render_token (line 273) | def render_token(content: Any) -> None:
  function render_time_info (line 286) | def render_time_info(content: Any) -> None:
  function render_generic (line 294) | def render_generic(content: Any) -> None:
  function render_summary (line 303) | def render_summary(summary: dict) -> None:

FILE: rdagent/app/rl/ui/data_loader.py
  class Event (line 21) | class Event:
    method time_str (line 35) | def time_str(self) -> str:
  class Loop (line 40) | class Loop:
  class Session (line 51) | class Session:
  function extract_loop_id (line 58) | def extract_loop_id(tag: str) -> int | None:
  function extract_stage (line 63) | def extract_stage(tag: str) -> str:
  function get_valid_sessions (line 75) | def get_valid_sessions(log_folder: Path) -> list[str]:
  function parse_event (line 85) | def parse_event(tag: str, content: Any, timestamp: datetime) -> Event | ...
  function load_session (line 237) | def load_session(log_path: Path) -> Session:
  function get_summary (line 285) | def get_summary(session: Session) -> dict:

FILE: rdagent/app/rl/ui/rl_summary.py
  function is_valid_task (line 13) | def is_valid_task(task_path: Path) -> bool:
  function get_loop_dirs (line 18) | def get_loop_dirs(task_path: Path) -> list[Path]:
  function get_loop_status (line 24) | def get_loop_status(task_path: Path, loop_id: int) -> tuple[str, bool | ...
  function get_max_loops (line 64) | def get_max_loops(job_path: Path) -> int:
  function get_job_summary_df (line 74) | def get_job_summary_df(job_path: Path) -> tuple[pd.DataFrame, pd.DataFra...
  function style_status_cell (line 119) | def style_status_cell(val: str, decision: bool | None = None) -> str:
  function style_df_with_decisions (line 136) | def style_df_with_decisions(df: pd.DataFrame, decisions_df: pd.DataFrame):
  function render_job_summary (line 152) | def render_job_summary(job_path: Path, is_root: bool = False) -> None:

FILE: rdagent/app/utils/ape.py
  function get_llm_qa (line 11) | def get_llm_qa(file_path):

FILE: rdagent/app/utils/health_check.py
  function check_docker_status (line 14) | def check_docker_status() -> None:
  function is_port_in_use (line 32) | def is_port_in_use(port):
  function check_and_list_free_ports (line 37) | def check_and_list_free_ports(start_port=19899, max_ports=10) -> None:
  function test_chat (line 51) | def test_chat(chat_model, chat_api_key, chat_api_base):
  function test_embedding (line 78) | def test_embedding(embedding_model, embedding_api_key, embedding_api_base):
  function env_check (line 94) | def env_check():
  function health_check (line 135) | def health_check(

FILE: rdagent/app/utils/info.py
  function sys_info (line 14) | def sys_info():
  function python_info (line 27) | def python_info():
  function docker_info (line 34) | def docker_info():
  function rdagent_info (line 52) | def rdagent_info():
  function collect_info (line 80) | def collect_info():

FILE: rdagent/app/utils/ws.py
  function run (line 13) | def run(competition: str, cmd: str, local_path: str = "./", mount_path: ...

FILE: rdagent/app/utils/ws_ft.py
  function run (line 13) | def run(

FILE: rdagent/components/agent/base.py
  class BaseAgent (line 12) | class BaseAgent:
    method __init__ (line 15) | def __init__(self, system_prompt: str, toolsets: list[str]): ...
    method query (line 18) | def query(self, query: str) -> str: ...
  class PAIAgent (line 21) | class PAIAgent(BaseAgent):
    method __init__ (line 29) | def __init__(
    method _run_query (line 56) | def _run_query(self, query: str) -> str:
    method query (line 64) | def query(self, query: str) -> str:

FILE: rdagent/components/agent/context7/__init__.py
  class Agent (line 11) | class Agent(PAIAgent):
    method __init__ (line 16) | def __init__(self):
    method _build_enhanced_query (line 25) | def _build_enhanced_query(self, error_message: str, full_code: Optiona...
    method query (line 46) | def query(self, query: str) -> str:

FILE: rdagent/components/agent/context7/conf.py
  class Settings (line 17) | class Settings(BaseSettings):

FILE: rdagent/components/agent/rag/__init__.py
  class Agent (line 8) | class Agent(PAIAgent):
    method __init__ (line 13) | def __init__(self, system_prompt: str | None = None):

FILE: rdagent/components/agent/rag/conf.py
  class Settings (line 10) | class Settings(BaseSettings):

FILE: rdagent/components/benchmark/conf.py
  class BenchmarkSettings (line 10) | class BenchmarkSettings(ExtendedBaseSettings):
    class Config (line 11) | class Config:

FILE: rdagent/components/benchmark/eval_method.py
  class TestCase (line 31) | class TestCase:
    method __init__ (line 32) | def __init__(
  class TestCases (line 41) | class TestCases:
    method __init__ (line 42) | def __init__(self, test_case_l: list[TestCase] = []):
    method __getitem__ (line 46) | def __getitem__(self, item):
    method __len__ (line 49) | def __len__(self):
    method get_exp (line 52) | def get_exp(self):
    method target_task (line 56) | def target_task(self):
    method ground_truth (line 60) | def ground_truth(self):
  class BaseEval (line 64) | class BaseEval:
    method __init__ (line 69) | def __init__(
    method load_cases_to_eval (line 90) | def load_cases_to_eval(
    method eval_case (line 105) | def eval_case(
  class FactorImplementEval (line 140) | class FactorImplementEval(BaseEval):
    method __init__ (line 141) | def __init__(
    method develop (line 160) | def develop(self):
    method eval (line 181) | def eval(self, gen_factor_l_all_rounds):
    method summarize_res (line 200) | def summarize_res(res: EVAL_RES) -> pd.DataFrame:

FILE: rdagent/components/benchmark/utils.py
  function _guess_dataset_var (line 13) | def _guess_dataset_var(mod_path: str) -> str:
  function build_dataset_imports_explicit (line 30) | def build_dataset_imports_explicit(dataset_imports: str | Iterable[str])...

FILE: rdagent/components/coder/CoSTEER/__init__.py
  class CoSTEER (line 20) | class CoSTEER(Developer[Experiment]):
    method __init__ (line 21) | def __init__(
    method get_develop_max_seconds (line 69) | def get_develop_max_seconds(self) -> int | None:
    method _get_last_fb (line 76) | def _get_last_fb(self) -> CoSTEERMultiFeedback:
    method should_use_new_evo (line 82) | def should_use_new_evo(self, base_fb: CoSTEERMultiFeedback | None, new...
    method develop (line 93) | def develop(self, exp: Experiment) -> Experiment:
    method _exp_postprocess_by_feedback (line 157) | def _exp_postprocess_by_feedback(self, evo: Experiment, feedback: CoST...

FILE: rdagent/components/coder/CoSTEER/config.py
  class CoSTEERSettings (line 6) | class CoSTEERSettings(ExtendedBaseSettings):
    class Config (line 9) | class Config:

FILE: rdagent/components/coder/CoSTEER/evaluators.py
  class CoSTEERSingleFeedback (line 32) | class CoSTEERSingleFeedback(Feedback):
    method val_and_update_init_dict (line 54) | def val_and_update_init_dict(data: dict) -> dict:
    method merge (line 86) | def merge(cls, feedback_li: list["CoSTEERSingleFeedback"]) -> "CoSTEER...
    method __str__ (line 110) | def __str__(self) -> str:
    method __bool__ (line 121) | def __bool__(self):
  class CoSTEERSingleFeedbackDeprecated (line 125) | class CoSTEERSingleFeedbackDeprecated(CoSTEERSingleFeedback):
    method __init__ (line 128) | def __init__(
    method execution (line 155) | def execution(self):
    method execution (line 159) | def execution(self, value):
    method return_checking (line 163) | def return_checking(self):
    method return_checking (line 169) | def return_checking(self, value):
    method code (line 176) | def code(self):
    method code (line 180) | def code(self, value):
    method __str__ (line 183) | def __str__(self) -> str:
  class CoSTEERMultiFeedback (line 199) | class CoSTEERMultiFeedback(Feedback):
    method __init__ (line 202) | def __init__(self, feedback_list: List[CoSTEERSingleFeedback]) -> None:
    method __getitem__ (line 205) | def __getitem__(self, index: int) -> CoSTEERSingleFeedback:
    method __len__ (line 208) | def __len__(self) -> int:
    method append (line 211) | def append(self, feedback: CoSTEERSingleFeedback) -> None:
    method __iter__ (line 214) | def __iter__(self):
    method is_acceptable (line 217) | def is_acceptable(self) -> bool:
    method finished (line 220) | def finished(self) -> bool:
    method __bool__ (line 227) | def __bool__(self) -> bool:
  class CoSTEEREvaluator (line 231) | class CoSTEEREvaluator(Evaluator):
    method __init__ (line 232) | def __init__(
    method evaluate (line 243) | def evaluate(
  class CoSTEERMultiEvaluator (line 253) | class CoSTEERMultiEvaluator(RAGEvaluator):
    method __init__ (line 256) | def __init__(self, single_evaluator: CoSTEEREvaluator | list[CoSTEEREv...
    method evaluate_iter (line 261) | def evaluate_iter(

FILE: rdagent/components/coder/CoSTEER/evolvable_subjects.py
  class EvolvingItem (line 6) | class EvolvingItem(Experiment, EvolvableSubjects):
    method __init__ (line 11) | def __init__(
    method from_experiment (line 28) | def from_experiment(cls, exp: Experiment) -> "EvolvingItem":

FILE: rdagent/components/coder/CoSTEER/evolving_strategy.py
  class MultiProcessEvolvingStrategy (line 22) | class MultiProcessEvolvingStrategy(EvolvingStrategy):
    method __init__ (line 25) | def __init__(self, scen: Scenario, settings: CoSTEERSettings, improve_...
    method implement_one_task (line 30) | def implement_one_task(
    method implement_func_list (line 61) | def implement_func_list(self) -> list[Callable]:
    method assign_code_list_to_evo (line 71) | def assign_code_list_to_evo(self, code_list: list[dict], evo: Evolving...
    method assign_code_list_to_evo (line 86) | def assign_code_list_to_evo(self, code_list: list[dict | None], evo) -...
    method evolve_iter (line 108) | def evolve_iter(

FILE: rdagent/components/coder/CoSTEER/knowledge_management.py
  class CoSTEERKnowledge (line 36) | class CoSTEERKnowledge(Knowledge):
    method __init__ (line 37) | def __init__(
    method get_implementation_and_feedback_str (line 47) | def get_implementation_and_feedback_str(self) -> str:
  class CoSTEERRAGStrategy (line 55) | class CoSTEERRAGStrategy(RAGStrategy):
    method __init__ (line 56) | def __init__(self, *args, dump_knowledge_base_path: Path = None, **kwa...
    method load_or_init_knowledge_base (line 60) | def load_or_init_knowledge_base(
    method dump_knowledge_base (line 82) | def dump_knowledge_base(self):
    method load_dumped_knowledge_base (line 91) | def load_dumped_knowledge_base(self, *args, **kwargs):
  class CoSTEERQueriedKnowledge (line 102) | class CoSTEERQueriedKnowledge(QueriedKnowledge):
    method __init__ (line 135) | def __init__(self, success_task_to_knowledge_dict: dict = {}, failed_t...
  class CoSTEERKnowledgeBaseV1 (line 140) | class CoSTEERKnowledgeBaseV1(EvolvingKnowledgeBase):
    method __init__ (line 141) | def __init__(self, path: str | Path = None) -> None:
    method query (line 148) | def query(self) -> CoSTEERQueriedKnowledge | None:
  class CoSTEERQueriedKnowledgeV1 (line 155) | class CoSTEERQueriedKnowledgeV1(CoSTEERQueriedKnowledge):
    method __init__ (line 156) | def __init__(
  class CoSTEERRAGStrategyV1 (line 168) | class CoSTEERRAGStrategyV1(CoSTEERRAGStrategy):
    method __init__ (line 171) | def __init__(self, settings: CoSTEERSettings, *args, **kwargs) -> None:
    method generate_knowledge (line 176) | def generate_knowledge(
    method query (line 219) | def query(
  class CoSTEERQueriedKnowledgeV2 (line 281) | class CoSTEERQueriedKnowledgeV2(CoSTEERQueriedKnowledgeV1):
    method __init__ (line 339) | def __init__(
  class CoSTEERRAGStrategyV2 (line 354) | class CoSTEERRAGStrategyV2(CoSTEERRAGStrategy):
    method __init__ (line 355) | def __init__(self, settings: CoSTEERSettings, *args, **kwargs) -> None:
    method generate_knowledge (line 360) | def generate_knowledge(
    method query (line 431) | def query(self, evo: EvolvableSubjects, evolving_trace: list[EvoStep])...
    method analyze_component (line 457) | def analyze_component(
    method analyze_error (line 488) | def analyze_error(
    method former_trace_query (line 530) | def former_trace_query(
    method component_query (line 593) | def component_query(
    method error_query (line 723) | def error_query(
  class CoSTEERKnowledgeBaseV2 (line 852) | class CoSTEERKnowledgeBaseV2(EvolvingKnowledgeBase):
    method __init__ (line 853) | def __init__(self, init_component_list=None, path: str | Path = None) ...
    method get_all_nodes_by_label (line 881) | def get_all_nodes_by_label(self, label: str) -> list[UndirectedNode]:
    method update_success_task (line 884) | def update_success_task(
    method query (line 926) | def query(self):
    method graph_get_node_by_content (line 929) | def graph_get_node_by_content(self, content: str) -> UndirectedNode:
    method graph_query_by_content (line 932) | def graph_query_by_content(
    method graph_query_by_node (line 974) | def graph_query_by_node(
    method graph_query_by_intersection (line 1009) | def graph_query_by_intersection(

FILE: rdagent/components/coder/CoSTEER/task.py
  class CoSTEERTask (line 4) | class CoSTEERTask(Task):
    method __init__ (line 5) | def __init__(self, base_code: str = None, *args, **kwargs) -> None:

FILE: rdagent/components/coder/data_science/conf.py
  class DSCoderCoSTEERSettings (line 16) | class DSCoderCoSTEERSettings(CoSTEERSettings):
    class Config (line 19) | class Config:
  function get_ds_env (line 41) | def get_ds_env(
  function get_clear_ws_cmd (line 78) | def get_clear_ws_cmd(stage: Literal["before_training", "before_inference...

FILE: rdagent/components/coder/data_science/ensemble/__init__.py
  class EnsembleMultiProcessEvolvingStrategy (line 43) | class EnsembleMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 44) | def implement_one_task(
    method assign_code_list_to_evo (line 127) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class EnsembleCoSTEER (line 144) | class EnsembleCoSTEER(DSCoSTEER):
    method __init__ (line 145) | def __init__(

FILE: rdagent/components/coder/data_science/ensemble/eval.py
  class EnsembleCoSTEEREvaluator (line 24) | class EnsembleCoSTEEREvaluator(CoSTEEREvaluator):
    method evaluate (line 25) | def evaluate(

FILE: rdagent/components/coder/data_science/ensemble/exp.py
  class EnsembleTask (line 12) | class EnsembleTask(CoSTEERTask):

FILE: rdagent/components/coder/data_science/ensemble/test.py
  function load_ensemble_spec (line 26) | def load_ensemble_spec():
  function develop_one_competition (line 32) | def develop_one_competition(competition: str):

FILE: rdagent/components/coder/data_science/feature/__init__.py
  class FeatureMultiProcessEvolvingStrategy (line 28) | class FeatureMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 29) | def implement_one_task(
    method assign_code_list_to_evo (line 101) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class FeatureCoSTEER (line 118) | class FeatureCoSTEER(DSCoSTEER):
    method __init__ (line 119) | def __init__(

FILE: rdagent/components/coder/data_science/feature/eval.py
  class FeatureCoSTEEREvaluator (line 19) | class FeatureCoSTEEREvaluator(CoSTEEREvaluator):
    method evaluate (line 20) | def evaluate(

FILE: rdagent/components/coder/data_science/feature/exp.py
  class FeatureTask (line 12) | class FeatureTask(CoSTEERTask):

FILE: rdagent/components/coder/data_science/feature/test.py
  function develop_one_competition (line 15) | def develop_one_competition(competition: str):  # -> experiment

FILE: rdagent/components/coder/data_science/model/__init__.py
  class ModelMultiProcessEvolvingStrategy (line 30) | class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 31) | def implement_one_task(
    method assign_code_list_to_evo (line 133) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class ModelCoSTEER (line 150) | class ModelCoSTEER(DSCoSTEER):
    method __init__ (line 151) | def __init__(

FILE: rdagent/components/coder/data_science/model/eval.py
  class ModelGeneralCaseSpecEvaluator (line 29) | class ModelGeneralCaseSpecEvaluator(CoSTEEREvaluator):
    method evaluate (line 38) | def evaluate(

FILE: rdagent/components/coder/data_science/model/exp.py
  class ModelTask (line 7) | class ModelTask(CoSTEERTask):
    method __init__ (line 8) | def __init__(
    method get_task_information (line 17) | def get_task_information(self):

FILE: rdagent/components/coder/data_science/model/test.py
  function develop_one_competition (line 19) | def develop_one_competition(competition: str):

FILE: rdagent/components/coder/data_science/pipeline/__init__.py
  class PipelineMultiProcessEvolvingStrategy (line 54) | class PipelineMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 55) | def implement_one_task(
    method assign_code_list_to_evo (line 116) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class PipelineCoSTEER (line 133) | class PipelineCoSTEER(DSCoSTEER):
    method __init__ (line 134) | def __init__(

FILE: rdagent/components/coder/data_science/pipeline/eval.py
  class DSCoderFeedback (line 33) | class DSCoderFeedback(CoSTEERSingleFeedback):
    method val_and_update_init_dict (line 43) | def val_and_update_init_dict(data: dict) -> dict:
    method __str__ (line 71) | def __str__(self) -> str:
    method merge (line 87) | def merge(cls, feedback_li: list[CoSTEERSingleFeedback]) -> "DSCoderFe...
  class PipelineCoSTEEREvaluator (line 123) | class PipelineCoSTEEREvaluator(CoSTEEREvaluator):
    method evaluate (line 125) | def evaluate(

FILE: rdagent/components/coder/data_science/pipeline/exp.py
  class PipelineTask (line 5) | class PipelineTask(CoSTEERTask):
    method __init__ (line 6) | def __init__(self, name: str = "Pipeline", package_info: str | None = ...

FILE: rdagent/components/coder/data_science/raw_data_loader/__init__.py
  class DataLoaderMultiProcessEvolvingStrategy (line 58) | class DataLoaderMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 59) | def implement_one_task(
    method assign_code_list_to_evo (line 181) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class DataLoaderCoSTEER (line 198) | class DataLoaderCoSTEER(DSCoSTEER):
    method __init__ (line 199) | def __init__(
    method develop (line 222) | def develop(self, exp):

FILE: rdagent/components/coder/data_science/raw_data_loader/eval.py
  class DataLoaderCoSTEEREvaluator (line 26) | class DataLoaderCoSTEEREvaluator(CoSTEEREvaluator):
    method evaluate (line 27) | def evaluate(

FILE: rdagent/components/coder/data_science/raw_data_loader/exp.py
  class DataLoaderTask (line 5) | class DataLoaderTask(CoSTEERTask):

FILE: rdagent/components/coder/data_science/raw_data_loader/test.py
  function develop_one_competition (line 15) | def develop_one_competition(competition: str):  # -> experiment

FILE: rdagent/components/coder/data_science/share/doc.py
  class DocDev (line 12) | class DocDev(Developer[Experiment]):
    method develop (line 17) | def develop(self, exp: Experiment) -> None:

FILE: rdagent/components/coder/data_science/share/ds_costeer.py
  class DSCoSTEER (line 4) | class DSCoSTEER(CoSTEER):
    method get_develop_max_seconds (line 5) | def get_develop_max_seconds(self) -> int | None:

FILE: rdagent/components/coder/data_science/share/eval.py
  class ModelDumpEvaluator (line 29) | class ModelDumpEvaluator(CoSTEEREvaluator):
    method __init__ (line 32) | def __init__(self, scen: Scenario, data_type: Literal["sample", "full"]):
    method evaluate (line 36) | def evaluate(

FILE: rdagent/components/coder/data_science/share/notebook.py
  class NotebookConverter (line 22) | class NotebookConverter:
    method validate_code_format (line 27) | def validate_code_format(self, code: str) -> str | None:
    method convert (line 41) | def convert(

FILE: rdagent/components/coder/data_science/share/util.py
  class CodeSection (line 9) | class CodeSection(TypedDict):
  function extract_function_body (line 20) | def extract_function_body(source_code: str, function_name: str) -> Optio...
  function split_sections (line 39) | def split_sections(
  function split_code_sections (line 79) | def split_code_sections(source_code: str) -> tuple[Optional[str], list[s...
  function split_output_sections (line 86) | def split_output_sections(stdout: str, known_sections: list[str]) -> tup...
  function extract_comment_under_first_print (line 94) | def extract_comment_under_first_print(source_code) -> tuple[Optional[str...
  function extract_first_section_name_from_code (line 133) | def extract_first_section_name_from_code(source_code):
  function extract_first_section_name_from_output (line 151) | def extract_first_section_name_from_output(stdout: str) -> Optional[str]:
  function is_function_called (line 161) | def is_function_called(source_code: str, func_name: str) -> bool:
  function remove_function (line 178) | def remove_function(source_code: str, function_name: str) -> str:
  function remove_main_block (line 194) | def remove_main_block(source_code: str) -> str:
  function extract_top_level_functions_with_decorators_and_comments (line 224) | def extract_top_level_functions_with_decorators_and_comments(
  function split_code_and_output_into_sections (line 300) | def split_code_and_output_into_sections(code: str, stdout: str) -> list[...

FILE: rdagent/components/coder/data_science/utils.py
  function remove_eda_part (line 4) | def remove_eda_part(stdout: str) -> str:

FILE: rdagent/components/coder/data_science/workflow/__init__.py
  class WorkflowMultiProcessEvolvingStrategy (line 26) | class WorkflowMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 27) | def implement_one_task(
    method assign_code_list_to_evo (line 94) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class WorkflowCoSTEER (line 111) | class WorkflowCoSTEER(DSCoSTEER):
    method __init__ (line 112) | def __init__(

FILE: rdagent/components/coder/data_science/workflow/eval.py
  class WorkflowGeneralCaseSpecEvaluator (line 27) | class WorkflowGeneralCaseSpecEvaluator(CoSTEEREvaluator):
    method evaluate (line 36) | def evaluate(

FILE: rdagent/components/coder/data_science/workflow/exp.py
  class WorkflowTask (line 12) | class WorkflowTask(CoSTEERTask):
    method __init__ (line 13) | def __init__(self, name: str = "Workflow", *args, **kwargs) -> None:

FILE: rdagent/components/coder/data_science/workflow/test.py
  function develop_one_competition (line 18) | def develop_one_competition(competition: str):

FILE: rdagent/components/coder/factor_coder/__init__.py
  class FactorCoSTEER (line 12) | class FactorCoSTEER(CoSTEER):
    method __init__ (line 13) | def __init__(
    method develop (line 25) | def develop(self, exp: Experiment) -> Experiment:

FILE: rdagent/components/coder/factor_coder/config.py
  class FactorCoSTEERSettings (line 10) | class FactorCoSTEERSettings(CoSTEERSettings):
  function get_factor_env (line 32) | def get_factor_env(

FILE: rdagent/components/coder/factor_coder/eva_utils.py
  class FactorEvaluator (line 16) | class FactorEvaluator:
    method __init__ (line 19) | def __init__(self, scen=None) -> None:
    method evaluate (line 23) | def evaluate(
    method _get_df (line 46) | def _get_df(self, gt_implementation: Workspace, implementation: Worksp...
    method __str__ (line 63) | def __str__(self) -> str:
  class FactorCodeEvaluator (line 67) | class FactorCodeEvaluator(FactorEvaluator):
    method evaluate (line 68) | def evaluate(
  class FactorInfEvaluator (line 120) | class FactorInfEvaluator(FactorEvaluator):
    method evaluate (line 121) | def evaluate(
  class FactorSingleColumnEvaluator (line 142) | class FactorSingleColumnEvaluator(FactorEvaluator):
    method evaluate (line 143) | def evaluate(
  class FactorOutputFormatEvaluator (line 163) | class FactorOutputFormatEvaluator(FactorEvaluator):
    method evaluate (line 164) | def evaluate(
  class FactorDatetimeDailyEvaluator (line 217) | class FactorDatetimeDailyEvaluator(FactorEvaluator):
    method evaluate (line 218) | def evaluate(
  class FactorRowCountEvaluator (line 247) | class FactorRowCountEvaluator(FactorEvaluator):
    method evaluate (line 248) | def evaluate(
  class FactorIndexEvaluator (line 271) | class FactorIndexEvaluator(FactorEvaluator):
    method evaluate (line 272) | def evaluate(
  class FactorMissingValuesEvaluator (line 296) | class FactorMissingValuesEvaluator(FactorEvaluator):
    method evaluate (line 297) | def evaluate(
  class FactorEqualValueRatioEvaluator (line 317) | class FactorEqualValueRatioEvaluator(FactorEvaluator):
    method evaluate (line 318) | def evaluate(
  class FactorCorrelationEvaluator (line 348) | class FactorCorrelationEvaluator(FactorEvaluator):
    method __init__ (line 349) | def __init__(self, hard_check: bool, *args, **kwargs) -> None:
    method evaluate (line 353) | def evaluate(
  class FactorValueEvaluator (line 389) | class FactorValueEvaluator(FactorEvaluator):
    method evaluate (line 390) | def evaluate(
  class FactorFinalDecisionEvaluator (line 478) | class FactorFinalDecisionEvaluator(FactorEvaluator):
    method evaluate (line 479) | def evaluate(

FILE: rdagent/components/coder/factor_coder/evaluators.py
  class FactorEvaluatorForCoder (line 20) | class FactorEvaluatorForCoder(CoSTEEREvaluator):
    method __init__ (line 25) | def __init__(self, *args, **kwargs) -> None:
    method evaluate (line 31) | def evaluate(
  function shorten_prompt (line 124) | def shorten_prompt(tpl: str, render_kwargs: dict, shorten_key: str, max_...

FILE: rdagent/components/coder/factor_coder/evolving_strategy.py
  class FactorMultiProcessEvolvingStrategy (line 23) | class FactorMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method __init__ (line 24) | def __init__(self, *args, **kwargs) -> None:
    method error_summary (line 29) | def error_summary(
    method implement_one_task (line 60) | def implement_one_task(
    method assign_code_list_to_evo (line 166) | def assign_code_list_to_evo(self, code_list, evo):

FILE: rdagent/components/coder/factor_coder/factor.py
  class FactorTask (line 20) | class FactorTask(CoSTEERTask):
    method __init__ (line 23) | def __init__(
    method factor_description (line 44) | def factor_description(self):
    method get_task_information (line 48) | def get_task_information(self):
    method get_task_brief_information (line 54) | def get_task_brief_information(self):
    method get_task_information_and_implementation_result (line 60) | def get_task_information_and_implementation_result(self):
    method from_dict (line 70) | def from_dict(dict):
    method __repr__ (line 73) | def __repr__(self) -> str:
  class FactorFBWorkspace (line 77) | class FactorFBWorkspace(FBWorkspace):
    method __init__ (line 90) | def __init__(
    method hash_func (line 99) | def hash_func(self, data_type: str = "Debug") -> str:
    method execute (line 107) | def execute(self, data_type: str = "Debug") -> Tuple[str, pd.DataFrame]:
    method __str__ (line 212) | def __str__(self) -> str:
    method __repr__ (line 217) | def __repr__(self) -> str:
    method from_folder (line 221) | def from_folder(task: FactorTask, path: Union[str, Path], **kwargs):

FILE: rdagent/components/coder/finetune/__init__.py
  class LLMFinetuneEvolvingStrategy (line 45) | class LLMFinetuneEvolvingStrategy(MultiProcessEvolvingStrategy):
    method __init__ (line 48) | def __init__(self, scen: Scenario, settings, *args, **kwargs):
    method implement_func_list (line 52) | def implement_func_list(self) -> list[Callable]:
    method implement_data (line 55) | def implement_data(
    method _get_dataset_info (line 153) | def _get_dataset_info(self, involving_datasets: list[str], datasets_pa...
    method implement_lf_config (line 245) | def implement_lf_config(
    method _generate_llamafactory_config_with_llm (line 288) | def _generate_llamafactory_config_with_llm(
  class LLMFinetuneCoSTEER (line 368) | class LLMFinetuneCoSTEER(CoSTEER):
    method __init__ (line 371) | def __init__(

FILE: rdagent/components/coder/finetune/conf.py
  function is_docker_env (line 27) | def is_docker_env(env: Env) -> bool:
  function get_workspace_prefix (line 32) | def get_workspace_prefix(env: Env) -> str:
  function get_data_processing_cache_key (line 52) | def get_data_processing_cache_key(local_path: str | Path) -> list[list[s...
  class FTPathConfig (line 70) | class FTPathConfig:
    method is_docker (line 85) | def is_docker(self) -> bool:
    method models (line 92) | def models(self) -> str:
    method datasets (line 99) | def datasets(self) -> str:
    method workspace (line 106) | def workspace(self) -> str:
    method deepspeed (line 111) | def deepspeed(self) -> str:
  class FTCoderCoSTEERSettings (line 126) | class FTCoderCoSTEERSettings(CoSTEERSettings):
    class Config (line 129) | class Config:
  function _get_standard_ft_volumes (line 142) | def _get_standard_ft_volumes() -> dict:
  function get_ft_env (line 168) | def get_ft_env(
  function get_data_processing_env (line 231) | def get_data_processing_env(
  function clear_workspace (line 273) | def clear_workspace(workspace: FBWorkspace, env: Env) -> None:
  function get_benchmark_env (line 308) | def get_benchmark_env(
  function inject_data_stats (line 368) | def inject_data_stats(implementation: FBWorkspace, data: list, stdout: s...

FILE: rdagent/components/coder/finetune/eval.py
  class FTDataEvaluator (line 42) | class FTDataEvaluator(CoSTEEREvaluator):
    method evaluate (line 51) | def evaluate(
    method _generate_llm_feedback (line 134) | def _generate_llm_feedback(
    method _validate_data_json (line 207) | def _validate_data_json(self, data_json_path: Path) -> dict:
    method _update_dataset_info (line 246) | def _update_dataset_info(self, implementation: FBWorkspace, sample_cou...
    method _sample_data (line 271) | def _sample_data(self, data: list, n: int = 5) -> list:
    method _analyze_data_quality (line 277) | def _analyze_data_quality(self, data: list) -> dict:
  class FTCoderEvaluator (line 307) | class FTCoderEvaluator(CoSTEEREvaluator):
    method __init__ (line 310) | def __init__(self, *args, **kwargs):
    method evaluate (line 313) | def evaluate(

FILE: rdagent/components/coder/finetune/exp.py
  class FTTask (line 12) | class FTTask(CoSTEERTask):
    method __init__ (line 15) | def __init__(
    method get_task_information (line 31) | def get_task_information(self) -> str:

FILE: rdagent/components/coder/finetune/unified_validator.py
  class ValidationResult (line 43) | class ValidationResult:
  class LLMConfigValidator (line 54) | class LLMConfigValidator:
    method __init__ (line 63) | def __init__(self):
    method validate_and_test (line 66) | def validate_and_test(self, config_yaml: str, workspace: FBWorkspace, ...
    method _filter_parameters (line 89) | def _filter_parameters(self, config_yaml: str) -> tuple[str, List[str]]:
    method _inject_required_parameters (line 114) | def _inject_required_parameters(self, config_yaml: str) -> str:
    method _get_supported_parameters (line 128) | def _get_supported_parameters(self) -> Set[str]:
    method _parse_execution_log (line 158) | def _parse_execution_log(self, stdout: str, exit_code: int, failed_sta...
    method _run_micro_batch_test (line 256) | def _run_micro_batch_test(self, config_yaml: str, workspace: FBWorkspa...

FILE: rdagent/components/coder/model_coder/__init__.py
  class ModelCoSTEER (line 11) | class ModelCoSTEER(CoSTEER):
    method __init__ (line 12) | def __init__(

FILE: rdagent/components/coder/model_coder/benchmark/eval.py
  function get_data_conf (line 7) | def get_data_conf(init_val):
  class ModelImpValEval (line 17) | class ModelImpValEval:
    method evaluate (line 35) | def evaluate(self, gt: ModelFBWorkspace, gen: ModelFBWorkspace):

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/A-DGN.py
  class AntiSymmetricConv (line 13) | class AntiSymmetricConv(torch.nn.Module):
    method __init__ (line 54) | def __init__(
    method reset_parameters (line 87) | def reset_parameters(self):
    method forward (line 93) | def forward(self, x: Tensor, edge_index: Adj, *args, **kwargs) -> Tensor:
    method __repr__ (line 111) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/dirgnn.py
  class DirGNNConv (line 8) | class DirGNNConv(torch.nn.Module):
    method __init__ (line 26) | def __init__(
    method reset_parameters (line 54) | def reset_parameters(self):
    method forward (line 61) | def forward(self, x: Tensor, edge_index: Tensor) -> Tensor:
    method __repr__ (line 73) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/gpsconv.py
  class GPSConv (line 16) | class GPSConv(torch.nn.Module):
    method __init__ (line 59) | def __init__(
    method reset_parameters (line 116) | def reset_parameters(self):
    method forward (line 129) | def forward(
    method __repr__ (line 178) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/linkx.py
  class SparseLinear (line 13) | class SparseLinear(MessagePassing):
    method __init__ (line 14) | def __init__(self, in_channels: int, out_channels: int, bias: bool = T...
    method reset_parameters (line 27) | def reset_parameters(self):
    method forward (line 31) | def forward(
    method message (line 44) | def message(self, weight_j: Tensor, edge_weight: OptTensor) -> Tensor:
    method message_and_aggregate (line 50) | def message_and_aggregate(self, adj_t: Adj, weight: Tensor) -> Tensor:
  class LINKX (line 54) | class LINKX(torch.nn.Module):
    method __init__ (line 88) | def __init__(
    method reset_parameters (line 127) | def reset_parameters(self):
    method forward (line 139) | def forward(
    method __repr__ (line 162) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/pmlp.py
  class PMLP (line 10) | class PMLP(torch.nn.Module):
    method __init__ (line 30) | def __init__(
    method reset_parameters (line 68) | def reset_parameters(self):
    method forward (line 75) | def forward(
    method __repr__ (line 98) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/benchmark/gt_code/visnet.py
  class CosineCutoff (line 12) | class CosineCutoff(torch.nn.Module):
    method __init__ (line 28) | def __init__(self, cutoff: float) -> None:
    method forward (line 32) | def forward(self, distances: Tensor) -> Tensor:
  class ExpNormalSmearing (line 48) | class ExpNormalSmearing(torch.nn.Module):
    method __init__ (line 64) | def __init__(
    method _initial_params (line 86) | def _initial_params(self) -> Tuple[Tensor, Tensor]:
    method reset_parameters (line 93) | def reset_parameters(self):
    method forward (line 99) | def forward(self, dist: Tensor) -> Tensor:
  class Sphere (line 110) | class Sphere(torch.nn.Module):
    method __init__ (line 124) | def __init__(self, lmax: int = 2) -> None:
    method forward (line 128) | def forward(self, edge_vec: Tensor) -> Tensor:
    method _spherical_harmonics (line 142) | def _spherical_harmonics(
  class VecLayerNorm (line 188) | class VecLayerNorm(torch.nn.Module):
    method __init__ (line 203) | def __init__(
    method reset_parameters (line 223) | def reset_parameters(self):
    method max_min_norm (line 227) | def max_min_norm(self, vec: Tensor) -> Tensor:
    method forward (line 258) | def forward(self, vec: Tensor) -> Tensor:
  class Distance (line 279) | class Distance(torch.nn.Module):
    method __init__ (line 296) | def __init__(
    method forward (line 307) | def forward(
  class NeighborEmbedding (line 344) | class NeighborEmbedding(MessagePassing):
    method __init__ (line 358) | def __init__(
    method reset_parameters (line 373) | def reset_parameters(self):
    method forward (line 381) | def forward(
    method message (line 416) | def message(self, x_j: Tensor, W: Tensor) -> Tensor:
  class EdgeEmbedding (line 420) | class EdgeEmbedding(torch.nn.Module):
    method __init__ (line 431) | def __init__(self, num_rbf: int, hidden_channels: int) -> None:
    method reset_parameters (line 436) | def reset_parameters(self):
    method forward (line 441) | def forward(
  class ViS_MP (line 462) | class ViS_MP(MessagePassing):
    method __init__ (line 482) | def __init__(
    method vector_rejection (line 536) | def vector_rejection(vec: Tensor, d_ij: Tensor) -> Tensor:
    method reset_parameters (line 546) | def reset_parameters(self):
    method forward (line 573) | def forward(
    method message (line 624) | def message(
    method edge_update (line 638) | def edge_update(self, vec_i: Tensor, vec_j: Tensor, d_ij: Tensor, f_ij...
    method aggregate (line 645) | def aggregate(
  class ViS_MP_Vertex (line 658) | class ViS_MP_Vertex(ViS_MP):
    method __init__ (line 678) | def __init__(
    method reset_parameters (line 696) | def reset_parameters(self):
    method edge_update (line 706) | def edge_update(self, vec_i: Tensor, vec_j: Tensor, d_ij: Tensor, f_ij...
  class ViSNetBlock (line 720) | class ViSNetBlock(torch.nn.Module):
    method __init__ (line 752) | def __init__(
    method reset_parameters (line 811) | def reset_parameters(self):
    method forward (line 822) | def forward(
  class GatedEquivariantBlock (line 866) | class GatedEquivariantBlock(torch.nn.Module):
    method __init__ (line 884) | def __init__(
    method reset_parameters (line 910) | def reset_parameters(self):
    method forward (line 919) | def forward(self, x: Tensor, v: Tensor) -> Tuple[Tensor, Tensor]:
  class EquivariantScalar (line 940) | class EquivariantScalar(torch.nn.Module):
    method __init__ (line 949) | def __init__(self, hidden_channels: int) -> None:
    method reset_parameters (line 969) | def reset_parameters(self):
    method pre_reduce (line 974) | def pre_reduce(self, x: Tensor, v: Tensor) -> Tensor:
  class Atomref (line 990) | class Atomref(torch.nn.Module):
    method __init__ (line 1000) | def __init__(
    method reset_parameters (line 1020) | def reset_parameters(self):
    method forward (line 1024) | def forward(self, x: Tensor, z: Tensor) -> Tensor:
  class ViSNet (line 1034) | class ViSNet(torch.nn.Module):
    method __init__ (line 1077) | def __init__(
    method reset_parameters (line 1124) | def reset_parameters(self):
    method forward (line 1131) | def forward(

FILE: rdagent/components/coder/model_coder/conf.py
  class ModelCoSTEERSettings (line 9) | class ModelCoSTEERSettings(CoSTEERSettings):
  function get_model_env (line 16) | def get_model_env(

FILE: rdagent/components/coder/model_coder/eva_utils.py
  function shape_evaluator (line 15) | def shape_evaluator(prediction: np.ndarray, target_shape: Tuple = None) ...
  function value_evaluator (line 32) | def value_evaluator(
  class ModelCodeEvaluator (line 52) | class ModelCodeEvaluator(CoSTEEREvaluator):
    method evaluate (line 53) | def evaluate(
  class ModelFinalEvaluator (line 105) | class ModelFinalEvaluator(CoSTEEREvaluator):
    method evaluate (line 106) | def evaluate(

FILE: rdagent/components/coder/model_coder/evaluators.py
  class ModelCoSTEEREvaluator (line 20) | class ModelCoSTEEREvaluator(CoSTEEREvaluator):
    method evaluate (line 21) | def evaluate(

FILE: rdagent/components/coder/model_coder/evolving_strategy.py
  class ModelMultiProcessEvolvingStrategy (line 20) | class ModelMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 21) | def implement_one_task(
    method assign_code_list_to_evo (line 82) | def assign_code_list_to_evo(self, code_list, evo):

FILE: rdagent/components/coder/model_coder/gt_code.py
  class AntiSymmetricConv (line 18) | class AntiSymmetricConv(torch.nn.Module):
    method __init__ (line 59) | def __init__(
    method reset_parameters (line 92) | def reset_parameters(self):
    method forward (line 98) | def forward(self, x: Tensor, edge_index: Adj, *args, **kwargs) -> Tensor:
    method __repr__ (line 116) | def __repr__(self) -> str:

FILE: rdagent/components/coder/model_coder/model.py
  class ModelTask (line 15) | class ModelTask(CoSTEERTask):
    method __init__ (line 16) | def __init__(
    method get_task_information (line 39) | def get_task_information(self):
    method get_task_brief_information (line 51) | def get_task_brief_information(self):
    method from_dict (line 62) | def from_dict(dict):
    method __repr__ (line 65) | def __repr__(self) -> str:
  class ModelFBWorkspace (line 69) | class ModelFBWorkspace(FBWorkspace):
    method hash_func (line 90) | def hash_func(
    method execute (line 105) | def execute(

FILE: rdagent/components/coder/model_coder/one_shot/__init__.py
  class ModelCodeWriter (line 12) | class ModelCodeWriter(Developer[ModelExperiment]):
    method develop (line 13) | def develop(self, exp: ModelExperiment) -> ModelExperiment:

FILE: rdagent/components/coder/model_coder/task_loader.py
  function extract_model_from_doc (line 20) | def extract_model_from_doc(doc_content: str) -> dict:
  function merge_file_to_model_dict_to_model_dict (line 71) | def merge_file_to_model_dict_to_model_dict(
  function extract_model_from_docs (line 92) | def extract_model_from_docs(docs_dict):
  class ModelExperimentLoaderFromDict (line 99) | class ModelExperimentLoaderFromDict(ModelTaskLoader):
    method load (line 100) | def load(self, model_dict: dict) -> QlibModelExperiment:
  class ModelExperimentLoaderFromPDFfiles (line 118) | class ModelExperimentLoaderFromPDFfiles(ModelTaskLoader):
    method load (line 120) | def load(self, file_or_folder_path: str) -> QlibModelExperiment:

FILE: rdagent/components/coder/rl/costeer.py
  class RLCoderCoSTEERSettings (line 23) | class RLCoderCoSTEERSettings(CoSTEERSettings):
  class RLEvolvingStrategy (line 29) | class RLEvolvingStrategy(EvolvingStrategy):
    method __init__ (line 32) | def __init__(self, scen: Scenario, settings: CoSTEERSettings):
    method evolve_iter (line 36) | def evolve_iter(
    method _generate_code (line 54) | def _generate_code(self, task: Task, evolving_trace: list[EvoStep] = [...
    method _mock_code (line 85) | def _mock_code(self) -> dict[str, str]:
  class RLCoderEvaluator (line 98) | class RLCoderEvaluator:
    method __init__ (line 101) | def __init__(self, scen: Scenario) -> None:
    method evaluate (line 104) | def evaluate(
  class RLCoSTEER (line 121) | class RLCoSTEER(CoSTEER):
    method __init__ (line 124) | def __init__(self, scen: Scenario, *args, **kwargs) -> None:

FILE: rdagent/components/document_reader/document_reader.py
  function load_documents_by_langchain (line 20) | def load_documents_by_langchain(path: str) -> list:
  function process_documents_by_langchain (line 36) | def process_documents_by_langchain(docs: list[Document]) -> dict[str, str]:
  function load_and_process_pdfs_by_langchain (line 63) | def load_and_process_pdfs_by_langchain(path: str) -> dict[str, str]:
  function load_and_process_one_pdf_by_azure_document_intelligence (line 67) | def load_and_process_one_pdf_by_azure_document_intelligence(
  function load_and_process_pdfs_by_azure_document_intelligence (line 87) | def load_and_process_pdfs_by_azure_document_intelligence(path: Path) -> ...
  function extract_first_page_screenshot_from_pdf (line 112) | def extract_first_page_screenshot_from_pdf(pdf_path: str) -> Image:

FILE: rdagent/components/interactor/__init__.py
  class SkipInteractor (line 6) | class SkipInteractor(Interactor[ASpecificExp]):
    method interact (line 8) | def interact(self, exp: ASpecificExp, trace: Trace) -> ASpecificExp:

FILE: rdagent/components/knowledge_management/graph.py
  class UndirectedNode (line 22) | class UndirectedNode(Node):
    method __init__ (line 23) | def __init__(self, content: str = "", label: str = "", embedding: Any ...
    method add_neighbor (line 29) | def add_neighbor(self, node: UndirectedNode) -> None:
    method remove_neighbor (line 33) | def remove_neighbor(self, node: UndirectedNode) -> None:
    method get_neighbors (line 38) | def get_neighbors(self) -> set[UndirectedNode]:
    method __str__ (line 41) | def __str__(self) -> str:
    method __repr__ (line 47) | def __repr__(self) -> str:
  class Graph (line 54) | class Graph(KnowledgeBase):
    method __init__ (line 59) | def __init__(self, path: str | Path | None = None) -> None:
    method size (line 63) | def size(self) -> int:
    method get_node (line 66) | def get_node(self, node_id: str) -> Node | None:
    method add_node (line 69) | def add_node(self, **kwargs: Any) -> NoReturn:
    method get_all_nodes (line 72) | def get_all_nodes(self) -> list[Node]:
    method get_all_nodes_by_label_list (line 75) | def get_all_nodes_by_label_list(self, label_list: list[str]) -> list[N...
    method find_node (line 78) | def find_node(self, content: str, label: str) -> Node | None:
    method batch_embedding (line 85) | def batch_embedding(nodes: list[Node]) -> list[Node]:
    method __str__ (line 104) | def __str__(self) -> str:
  class UndirectedGraph (line 108) | class UndirectedGraph(Graph):
    method __init__ (line 113) | def __init__(self, path: str | Path | None = None) -> None:
    method __str__ (line 117) | def __str__(self) -> str:
    method add_node (line 120) | def add_node(
    method add_nodes (line 169) | def add_nodes(self, node: UndirectedNode, neighbors: list[UndirectedNo...
    method get_node (line 176) | def get_node(self, node_id: str) -> UndirectedNode:
    method get_node_by_content (line 179) | def get_node_by_content(self, content: str) -> UndirectedNode | None:
    method get_nodes_within_steps (line 197) | def get_nodes_within_steps(
    method get_nodes_intersection (line 235) | def get_nodes_intersection(
    method semantic_search (line 275) | def semantic_search(
    method clear (line 314) | def clear(self) -> None:
    method query_by_node (line 318) | def query_by_node(
    method query_by_content (line 356) | def query_by_content(
    method intersection (line 430) | def intersection(nodes1: list[UndirectedNode], nodes2: list[Undirected...
    method different (line 434) | def different(nodes1: list[UndirectedNode], nodes2: list[UndirectedNod...
    method cal_distance (line 438) | def cal_distance(node1: UndirectedNode, node2: UndirectedNode) -> float:
    method filter_label (line 442) | def filter_label(nodes: list[UndirectedNode], labels: list[str]) -> li...
  function graph_to_edges (line 446) | def graph_to_edges(graph: dict[str, list[str]]) -> list[tuple[str, str]]:
  function assign_random_coordinate_to_node (line 458) | def assign_random_coordinate_to_node(
  function assign_isometric_coordinate_to_node (line 472) | def assign_isometric_coordinate_to_node(
  function curly_node_coordinate (line 488) | def curly_node_coordinate(

FILE: rdagent/components/knowledge_management/vector_base.py
  class KnowledgeMetaData (line 13) | class KnowledgeMetaData:
    method __init__ (line 14) | def __init__(self, content: str = "", label: str = None, embedding=Non...
    method split_into_trunk (line 22) | def split_into_trunk(self, size: int = 1000, overlap: int = 0):
    method create_embedding (line 40) | def create_embedding(self):
    method from_dict (line 50) | def from_dict(self, data: dict):
    method __repr__ (line 55) | def __repr__(self):
  function contents_to_documents (line 62) | def contents_to_documents(contents: List[str], label: str = None) -> Lis...
  class VectorBase (line 72) | class VectorBase(KnowledgeBase):
    method add (line 77) | def add(self, document: Union[Document, List[Document]]):
    method search (line 90) | def search(self, content: str, topk_k: int | None = None, similarity_t...
  class PDVectorBase (line 106) | class PDVectorBase(VectorBase):
    method __init__ (line 111) | def __init__(self, path: Union[str, Path] = None):
    method shape (line 115) | def shape(self):
    method add (line 118) | def add(self, document: Union[Document, List[Document]]):
    method search (line 158) | def search(

FILE: rdagent/components/loader/experiment_loader.py
  class FactorExperimentLoader (line 5) | class FactorExperimentLoader(Loader[FactorExperiment]):
  class ModelExperimentLoader (line 9) | class ModelExperimentLoader(Loader[FactorExperiment]):

FILE: rdagent/components/loader/task_loader.py
  class FactorTaskLoader (line 10) | class FactorTaskLoader(Loader[FactorTask]):
  class ModelTaskLoader (line 14) | class ModelTaskLoader(Loader[ModelTask]):
  class ModelTaskLoaderJson (line 18) | class ModelTaskLoaderJson(ModelTaskLoader):
    method __init__ (line 44) | def __init__(self, json_uri: str) -> None:
    method load (line 48) | def load(self, *argT, **kwargs) -> Sequence[ModelTask]:
  class ModelWsLoader (line 85) | class ModelWsLoader(WsLoader[ModelTask, ModelFBWorkspace]):
    method __init__ (line 86) | def __init__(self, path: Path) -> None:
    method load (line 89) | def load(self, task: ModelTask) -> ModelFBWorkspace:

FILE: rdagent/components/proposal/__init__.py
  class LLMHypothesisGen (line 18) | class LLMHypothesisGen(HypothesisGen):
    method __init__ (line 19) | def __init__(self, scen: Scenario):
    method prepare_context (line 24) | def prepare_context(self, trace: Trace) -> Tuple[dict, bool]: ...
    method convert_response (line 27) | def convert_response(self, response: str) -> Hypothesis: ...
    method gen (line 29) | def gen(
  class FactorHypothesisGen (line 68) | class FactorHypothesisGen(LLMHypothesisGen):
    method __init__ (line 69) | def __init__(self, scen: Scenario):
  class ModelHypothesisGen (line 74) | class ModelHypothesisGen(LLMHypothesisGen):
    method __init__ (line 75) | def __init__(self, scen: Scenario):
  class FactorAndModelHypothesisGen (line 80) | class FactorAndModelHypothesisGen(LLMHypothesisGen):
    method __init__ (line 81) | def __init__(self, scen: Scenario):
  class LLMHypothesis2Experiment (line 86) | class LLMHypothesis2Experiment(Hypothesis2Experiment[Experiment]):
    method prepare_context (line 88) | def prepare_context(self, hypothesis: Hypothesis, trace: Trace) -> Tup...
    method convert_response (line 91) | def convert_response(self, response: str, hypothesis: Hypothesis, trac...
    method convert (line 94) | def convert(self, hypothesis: Hypothesis, trace: Trace) -> Experiment:
  class FactorHypothesis2Experiment (line 124) | class FactorHypothesis2Experiment(LLMHypothesis2Experiment):
    method __init__ (line 125) | def __init__(self):
  class ModelHypothesis2Experiment (line 130) | class ModelHypothesis2Experiment(LLMHypothesis2Experiment):
    method __init__ (line 131) | def __init__(self):
  class FactorAndModelHypothesis2Experiment (line 136) | class FactorAndModelHypothesis2Experiment(LLMHypothesis2Experiment):
    method __init__ (line 137) | def __init__(self):

FILE: rdagent/components/runner/__init__.py
  class CachedRunner (line 6) | class CachedRunner(Developer[ASpecificExp]):
    method get_cache_key (line 7) | def get_cache_key(self, exp: Experiment) -> str:
    method assign_cached_result (line 16) | def assign_cached_result(self, exp: Experiment, cached_res: Experiment...

FILE: rdagent/components/workflow/conf.py
  class BasePropSetting (line 4) | class BasePropSetting(ExtendedBaseSettings):

FILE: rdagent/components/workflow/rd_loop.py
  class RDLoop (line 31) | class RDLoop(LoopBase, metaclass=LoopMeta):
    method __init__ (line 33) | def __init__(self, PROP_SETTING: BasePropSetting):
    method _set_interactor (line 71) | def _set_interactor(self, user_request_q: Queue, user_response_q: Queue):
    method _init_base_features (line 75) | def _init_base_features(self, base_features_path: str | None):
    method _interact_init_params (line 112) | def _interact_init_params(self) -> None:
    method _interact_hypo (line 154) | def _interact_hypo(self, hypo: Hypothesis) -> Hypothesis:
    method _interact_feedback (line 169) | def _interact_feedback(self, feedback: HypothesisFeedback) -> Hypothes...
    method _propose (line 184) | def _propose(self):
    method _exp_gen (line 193) | def _exp_gen(self, hypothesis: Hypothesis):
    method direct_exp_gen (line 199) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 212) | def coding(self, prev_out: dict[str, Any]):
    method running (line 217) | def running(self, prev_out: dict[str, Any]):
    method feedback (line 222) | def feedback(self, prev_out: dict[str, Any]):
    method record (line 238) | def record(self, prev_out: dict[str, Any]):

FILE: rdagent/core/conf.py
  class ExtendedBaseSettings (line 13) | class ExtendedBaseSettings(BaseSettings):
    method settings_customise_sources (line 16) | def settings_customise_sources(
  class RDAgentSettings (line 46) | class RDAgentSettings(ExtendedBaseSettings):
    method get_max_parallel (line 92) | def get_max_parallel(self) -> int:
    method is_force_subproc (line 102) | def is_force_subproc(self) -> bool:

FILE: rdagent/core/developer.py
  class Developer (line 12) | class Developer(ABC, Generic[ASpecificExp]):
    method __init__ (line 13) | def __init__(self, scen: Scenario) -> None:
    method develop (line 17) | def develop(self, exp: ASpecificExp) -> ASpecificExp:  # TODO: remove ...

FILE: rdagent/core/evaluation.py
  class Feedback (line 8) | class Feedback:
    method is_acceptable (line 15) | def is_acceptable(self) -> bool:
    method finished (line 22) | def finished(self) -> bool:
    method __bool__ (line 29) | def __bool__(self) -> bool:
  class EvaluableObj (line 33) | class EvaluableObj:
  class Evaluator (line 42) | class Evaluator(ABC):
    method evaluate (line 53) | def evaluate(

FILE: rdagent/core/evolving_agent.py
  class EvoAgent (line 26) | class EvoAgent(ABC, Generic[ASpecificEvaluator, ASpecificEvolvableSubjec...
    method __init__ (line 28) | def __init__(self, max_loop: int, evolving_strategy: EvolvingStrategy)...
    method multistep_evolve (line 33) | def multistep_evolve(
  class RAGEvaluator (line 43) | class RAGEvaluator(IterEvaluator):
    method evaluate_iter (line 46) | def evaluate_iter(
  class RAGEvoAgent (line 78) | class RAGEvoAgent(EvoAgent[RAGEvaluator, ASpecificEvolvableSubjects], Ge...
    method __init__ (line 80) | def __init__(
    method _get_overall_feedback (line 120) | def _get_overall_feedback(
    method multistep_evolve (line 140) | def multistep_evolve(

FILE: rdagent/core/evolving_framework.py
  class Knowledge (line 17) | class Knowledge:
  class QueriedKnowledge (line 21) | class QueriedKnowledge:
  class EvolvingKnowledgeBase (line 25) | class EvolvingKnowledgeBase(KnowledgeBase):
    method query (line 27) | def query(
  class EvolvableSubjects (line 33) | class EvolvableSubjects(EvaluableObj):
    method clone (line 36) | def clone(self) -> EvolvableSubjects:
  class EvoStep (line 44) | class EvoStep(Generic[ASpecificEvolvableSubjects]):
  class EvolvingStrategy (line 61) | class EvolvingStrategy(ABC, Generic[ASpecificEvolvableSubjects]):
    method __init__ (line 62) | def __init__(self, scen: Scenario) -> None:
    method evolve_iter (line 66) | def evolve_iter(
  class IterEvaluator (line 94) | class IterEvaluator(Evaluator):
    method evaluate (line 101) | def evaluate(self, eo: EvaluableObj) -> Feedback:
    method evaluate_iter (line 115) | def evaluate_iter(self) -> Generator[Feedback, EvaluableObj | None, Fe...
  class RAGStrategy (line 141) | class RAGStrategy(ABC, Generic[ASpecificEvolvableSubjects]):
    method __init__ (line 144) | def __init__(self, *args: Any, **kwargs: Any) -> None:
    method load_or_init_knowledge_base (line 148) | def load_or_init_knowledge_base(
    method query (line 156) | def query(
    method generate_knowledge (line 165) | def generate_knowledge(
    method dump_knowledge_base (line 179) | def dump_knowledge_base(self, *args: Any, **kwargs: Any) -> None:
    method load_dumped_knowledge_base (line 183) | def load_dumped_knowledge_base(self, *args: Any, **kwargs: Any) -> None:

FILE: rdagent/core/exception.py
  class WorkflowError (line 1) | class WorkflowError(Exception):
  class FormatError (line 7) | class FormatError(WorkflowError):
  class CodeBlockParseError (line 13) | class CodeBlockParseError(FormatError):
    method __init__ (line 16) | def __init__(self, message: str, content: str, language: str) -> None:
  class CoderError (line 23) | class CoderError(WorkflowError):
  class CodeFormatError (line 36) | class CodeFormatError(CoderError):
  class CustomRuntimeError (line 42) | class CustomRuntimeError(CoderError):
  class NoOutputError (line 48) | class NoOutputError(CoderError):
  class RunnerError (line 54) | class RunnerError(Exception):
  class KaggleError (line 67) | class KaggleError(Exception):
  class PolicyError (line 73) | class PolicyError(Exception):
  class EvaluatorDidNotTerminateError (line 79) | class EvaluatorDidNotTerminateError(RuntimeError):

FILE: rdagent/core/experiment.py
  class AbsTask (line 34) | class AbsTask(ABC):
    method __init__ (line 35) | def __init__(self, name: str, version: int = 1) -> None:
    method get_task_information (line 45) | def get_task_information(self) -> str:
  class UserInstructions (line 51) | class UserInstructions(list[str]):
    method __str__ (line 52) | def __str__(self) -> str:
  class Task (line 58) | class Task(AbsTask):
    method __init__ (line 59) | def __init__(
    method get_task_information (line 70) | def get_task_information(self) -> str:
    method __repr__ (line 73) | def __repr__(self) -> str:
  class RunningInfo (line 82) | class RunningInfo:
  class Workspace (line 87) | class Workspace(ABC, Generic[ASpecificTask, ASpecificFeedback]):
    method __init__ (line 93) | def __init__(self, target_task: ASpecificTask | None = None) -> None:
    method execute (line 99) | def execute(self, *args: Any, **kwargs: Any) -> object | None:
    method copy (line 104) | def copy(self) -> Workspace:
    method all_codes (line 110) | def all_codes(self) -> str:
    method create_ws_ckp (line 117) | def create_ws_ckp(self) -> None:
    method recover_ws_ckp (line 123) | def recover_ws_ckp(self) -> None:
  class WsLoader (line 132) | class WsLoader(ABC, Generic[ASpecificTask, ASpecificWS]):
    method load (line 134) | def load(self, task: ASpecificTask) -> ASpecificWS:
  class FBWorkspace (line 139) | class FBWorkspace(Workspace):
    method __init__ (line 162) | def __init__(self, *args: Any, **kwargs: Any) -> None:
    method _format_code_dict (line 172) | def _format_code_dict(code_dict: dict[str, str]) -> str:
    method all_codes (line 182) | def all_codes(self) -> str:
    method get_codes (line 189) | def get_codes(self, pattern: str) -> str:
    method prepare (line 198) | def prepare(self) -> None:
    method link_all_files_in_folder_to_workspace (line 209) | def link_all_files_in_folder_to_workspace(data_path: Path, workspace_p...
    method inject_files (line 223) | def inject_files(self, **files: str) -> None:
    method remove_files (line 245) | def remove_files(self, file_names: str | list[str]) -> None:
    method get_files (line 257) | def get_files(self) -> list[Path]:
    method inject_code_from_folder (line 266) | def inject_code_from_folder(self, folder_path: Path) -> None:
    method inject_code_from_file_dict (line 275) | def inject_code_from_file_dict(self, workspace: FBWorkspace) -> None:
    method inject_from_workspace (line 283) | def inject_from_workspace(self, workspace: FBWorkspace) -> None:
    method copy (line 287) | def copy(self) -> FBWorkspace:
    method clear (line 293) | def clear(self) -> None:
    method before_execute (line 300) | def before_execute(self) -> None:
    method execute (line 307) | def execute(self, env: Env, entry: str) -> str:
    method run (line 314) | def run(self, env: Env, entry: str) -> EnvResult:
    method create_ws_ckp (line 324) | def create_ws_ckp(self) -> None:
    method recover_ws_ckp (line 350) | def recover_ws_ckp(self) -> None:
    method __str__ (line 380) | def __str__(self) -> str:
  class ExperimentPlan (line 390) | class ExperimentPlan(dict[str, Any]):
  class Experiment (line 396) | class Experiment(
    method __init__ (line 404) | def __init__(
    method set_user_instructions (line 446) | def set_user_instructions(self, user_instructions: UserInstructions | ...
    method result (line 461) | def result(self) -> object:
    method result (line 465) | def result(self, value: object) -> None:
    method create_ws_ckp (line 469) | def create_ws_ckp(self) -> None:
    method recover_ws_ckp (line 476) | def recover_ws_ckp(self) -> None:
  class Loader (line 495) | class Loader(ABC, Generic[TaskOrExperiment]):
    method load (line 497) | def load(self, *args: Any, **kwargs: Any) -> TaskOrExperiment:

FILE: rdagent/core/interactor.py
  class Interactor (line 13) | class Interactor(ABC, Generic[ASpecificExp]):
    method __init__ (line 14) | def __init__(self, scen: Scenario) -> None:
    method interact (line 18) | def interact(self, exp: ASpecificExp, trace: Trace | None = None) -> A...

FILE: rdagent/core/knowledge_base.py
  class KnowledgeBase (line 8) | class KnowledgeBase:
    method __init__ (line 9) | def __init__(self, path: str | Path | None = None) -> None:
    method load (line 13) | def load(self) -> None:
    method dump (line 22) | def dump(self) -> None:

FILE: rdagent/core/prompts.py
  class Prompts (line 8) | class Prompts(SingletonBaseClass, dict[str, str]):
    method __init__ (line 9) | def __init__(self, file_path: Path) -> None:

FILE: rdagent/core/proposal.py
  class Hypothesis (line 24) | class Hypothesis:
    method __init__ (line 32) | def __init__(
    method __str__ (line 48) | def __str__(self) -> str:
  class ExperimentFeedback (line 58) | class ExperimentFeedback(Feedback):
    method __init__ (line 59) | def __init__(
    method __bool__ (line 78) | def __bool__(self) -> bool:
    method __str__ (line 81) | def __str__(self) -> str:
    method from_exception (line 89) | def from_exception(cls, e: Exception) -> ExperimentFeedback:
  class HypothesisFeedback (line 96) | class HypothesisFeedback(ExperimentFeedback):
    method __init__ (line 97) | def __init__(
    method __str__ (line 122) | def __str__(self) -> str:
  class Trace (line 141) | class Trace(Generic[ASpecificScen, ASpecificKB]):
    method __init__ (line 146) | def __init__(self, scen: ASpecificScen, knowledge_base: ASpecificKB | ...
    method get_sota_hypothesis_and_experiment (line 178) | def get_sota_hypothesis_and_experiment(self) -> tuple[Hypothesis | Non...
    method is_selection_new_tree (line 187) | def is_selection_new_tree(self, selection: tuple[int, ...] | None = No...
    method get_current_selection (line 197) | def get_current_selection(self) -> tuple[int, ...]:
    method set_current_selection (line 200) | def set_current_selection(self, selection: tuple[int, ...]) -> None:
    method get_parent_exps (line 203) | def get_parent_exps(
    method exp2idx (line 219) | def exp2idx(self, exp: Experiment | list[Experiment]) -> int | list[in...
    method idx2exp (line 231) | def idx2exp(self, idx: int | list[int]) -> Experiment | list[Experiment]:
    method is_parent (line 237) | def is_parent(self, parent_idx: int, child_idx: int) -> bool:
    method get_parents (line 241) | def get_parents(self, child_idx: int) -> list[int]:
    method sync_dag_parent_and_hist (line 256) | def sync_dag_parent_and_hist(
    method get_children (line 286) | def get_children(self, parent_idx: int | None = None) -> list[NodeType]:
    method get_sota_experiment (line 298) | def get_sota_experiment(self, node_id: int | None = None) -> Experimen...
  class CheckpointSelector (line 321) | class CheckpointSelector:
    method get_selection (line 327) | def get_selection(self, trace: Trace) -> tuple[int, ...] | None:
  class SOTAexpSelector (line 343) | class SOTAexpSelector:
    method get_sota_exp_to_submit (line 349) | def get_sota_exp_to_submit(self, trace: Trace) -> Experiment | None:
  class ExpPlanner (line 355) | class ExpPlanner(ABC, Generic[ASpecificPlan]):
    method __init__ (line 361) | def __init__(self, scen: Scenario) -> None:
    method plan (line 365) | def plan(self, trace: Trace) -> ASpecificPlan:
  class ExpGen (line 372) | class ExpGen(ABC):
    method __init__ (line 374) | def __init__(self, scen: Scenario) -> None:
    method gen (line 378) | def gen(self, trace: Trace) -> Experiment:
    method async_gen (line 394) | async def async_gen(self, trace: Trace, loop: LoopBase) -> Experiment:
    method reset (line 405) | def reset(self) -> None:
  class HypothesisGen (line 414) | class HypothesisGen(ABC):
    method __init__ (line 416) | def __init__(self, scen: Scenario) -> None:
    method gen (line 420) | def gen(
  class Hypothesis2Experiment (line 437) | class Hypothesis2Experiment(ABC, Generic[ASpecificExp]):
    method convert (line 443) | def convert(self, hypothesis: Hypothesis, trace: Trace) -> ASpecificExp:
  class Experiment2Feedback (line 451) | class Experiment2Feedback(ABC):
    method __init__ (line 455) | def __init__(self, scen: Scenario) -> None:
    method generate_feedback (line 459) | def generate_feedback(

FILE: rdagent/core/scenario.py
  class Scenario (line 6) | class Scenario(ABC):
    method background (line 14) | def background(self) -> str:
    method get_source_data_desc (line 18) | def get_source_data_desc(self, task: Task | None = None) -> str:  # no...
    method source_data (line 27) | def source_data(self) -> str:
    method rich_style_description (line 39) | def rich_style_description(self) -> str:
    method get_scenario_all_desc (line 43) | def get_scenario_all_desc(
    method get_runtime_environment (line 56) | def get_runtime_environment(self) -> str:
    method experiment_setting (line 62) | def experiment_setting(self) -> str | None:

FILE: rdagent/core/utils.py
  class RDAgentException (line 20) | class RDAgentException(Exception):  # noqa: N818
  class SingletonBaseClass (line 24) | class SingletonBaseClass:
    method __new__ (line 32) | def __new__(cls, *args: Any, **kwargs: Any) -> Any:
    method __reduce__ (line 47) | def __reduce__(self) -> NoReturn:
  function parse_json (line 58) | def parse_json(response: str) -> Any:
  function similarity (line 67) | def similarity(text1: str, text2: str) -> int:
  function import_class (line 75) | def import_class(class_path: str) -> Any:
  class CacheSeedGen (line 91) | class CacheSeedGen:
    method __init__ (line 101) | def __init__(self) -> None:
    method set_seed (line 104) | def set_seed(self, seed: int) -> None:
    method get_next_seed (line 107) | def get_next_seed(self) -> int:
  function _subprocess_wrapper (line 115) | def _subprocess_wrapper(f: Callable, seed: int, args: list) -> Any:
  function multiprocessing_wrapper (line 124) | def multiprocessing_wrapper(func_calls: list[tuple[Callable, tuple]], n:...
  function cache_with_pickle (line 156) | def cache_with_pickle(hash_func: Callable, post_process_func: Callable |...

FILE: rdagent/log/base.py
  class Message (line 12) | class Message:
  class Storage (line 25) | class Storage:
    method log (line 43) | def log(
    method iter_msg (line 67) | def iter_msg(self) -> Generator[Message, None, None]:
    method truncate (line 74) | def truncate(self, time: datetime) -> None:
    method __str__ (line 80) | def __str__(self) -> str:
  class View (line 84) | class View:
    method display (line 93) | def display(self, s: Storage, watch: bool = False) -> None:

FILE: rdagent/log/conf.py
  class LogSettings (line 10) | class LogSettings(ExtendedBaseSettings):
    method set_ui_server_port (line 22) | def set_ui_server_port(self, port: int | None) -> None:
    method model_post_init (line 30) | def model_post_init(self, _context: Any, /) -> None:

FILE: rdagent/log/logger.py
  class RDAgentLog (line 20) | class RDAgentLog(SingletonBaseClass):
    method _configure_console_sinks (line 48) | def _configure_console_sinks(cls) -> None:
    method _tag (line 59) | def _tag(self) -> str:  # Get current tag
    method _tag (line 63) | def _tag(self, value: str) -> None:
    method __init__ (line 66) | def __init__(self) -> None:
    method refresh_storages_from_settings (line 76) | def refresh_storages_from_settings(self) -> None:
    method rebind_console_to_current_streams (line 82) | def rebind_console_to_current_streams(self) -> None:
    method tag (line 92) | def tag(self, tag: str) -> Generator[None, None, None]:
    method set_storages_path (line 106) | def set_storages_path(self, path: str | Path) -> None:
    method truncate_storages (line 113) | def truncate_storages(self, time: datetime) -> None:
    method get_pids (line 117) | def get_pids(self) -> str:
    method log_object (line 132) | def log_object(self, obj: object, *, tag: str = "") -> None:
    method _log (line 138) | def _log(self, level: str, msg: str, *, tag: str = "", raw: bool = Fal...
    method info (line 146) | def info(self, msg: str, *, tag: str = "", raw: bool = False) -> None:
    method warning (line 149) | def warning(self, msg: str, *, tag: str = "", raw: bool = False) -> None:
    method error (line 152) | def error(self, msg: str, *, tag: str = "", raw: bool = False) -> None:

FILE: rdagent/log/mle_summary.py
  function save_grade_info (line 25) | def save_grade_info(log_trace_path: Path):
  function save_all_grade_info (line 45) | def save_all_grade_info(log_folder: str | Path) -> None:
  function _get_loop_and_fn_after_hours (line 54) | def _get_loop_and_fn_after_hours(log_folder: Path, hours: int):
  function summarize_folder (line 74) | def summarize_folder(log_folder: Path, hours: int | None = None) -> None:
  function grade_summary (line 244) | def grade_summary(log_folder: str) -> None:

FILE: rdagent/log/server/app.py
  class _YellowWarningFormatter (line 30) | class _YellowWarningFormatter(logging.Formatter):
    method format (line 31) | def format(self, record: logging.LogRecord) -> str:
  function _configure_app_logger (line 37) | def _configure_app_logger() -> None:
  class RDAgentTask (line 52) | class RDAgentTask:
    method __init__ (line 53) | def __init__(
    method start (line 88) | def start(self) -> None:
    method is_alive (line 92) | def is_alive(self) -> bool:
    method get_end_code (line 95) | def get_end_code(self) -> int:
    method stop (line 100) | def stop(self) -> None:
    method _run (line 116) | def _run(self) -> None:
  function _drain_user_requests_into_messages (line 175) | def _drain_user_requests_into_messages(task: RDAgentTask) -> None:
  function favicon (line 202) | def favicon():
  function _normalize_static_request_path (line 206) | def _normalize_static_request_path(fn: str) -> str:
  function _get_or_create_task (line 213) | def _get_or_create_task(trace_id: str) -> RDAgentTask:
  function _resolve_stdout_path (line 230) | def _resolve_stdout_path(trace_id: str) -> Path | None:
  function read_trace (line 250) | def read_trace(log_path: Path, id: str = "") -> None:
  function update_trace (line 284) | def update_trace():
  function download_stdout_file (line 333) | def download_stdout_file():
  function upload_file (line 351) | def upload_file():
  function receive_msgs (line 454) | def receive_msgs():
  function submit_user_interaction_response (line 474) | def submit_user_interaction_response():
  function control_process (line 497) | def control_process():
  function test (line 537) | def test():
  function index (line 545) | def index():
  function server_static_files (line 552) | def server_static_files(fn):
  function main (line 556) | def main(port: int = 19899):

FILE: rdagent/log/server/debug_app.py
  function favicon (line 27) | def favicon():
  function update_trace (line 36) | def update_trace():
  function upload_file (line 74) | def upload_file():
  function receive_msgs (line 117) | def receive_msgs():
  function control_process (line 136) | def control_process():
  function test (line 150) | def test():
  function index (line 156) | def index():
  function server_static_files (line 163) | def server_static_files(fn):
  function main (line 167) | def main(port: int = 19899):

FILE: rdagent/log/storage.py
  function _remove_empty_dir (line 14) | def _remove_empty_dir(path: Path) -> None:
  class FileStorage (line 28) | class FileStorage(Storage):
    method __init__ (line 35) | def __init__(self, path: str | Path) -> None:
    method log (line 38) | def log(
    method iter_msg (line 79) | def iter_msg(self, tag: str | None = None, pattern: str | None = None)...
    method truncate (line 107) | def truncate(self, time: datetime) -> None:
    method __str__ (line 115) | def __str__(self) -> str:

FILE: rdagent/log/timer.py
  class RDAgentTimer (line 8) | class RDAgentTimer:
    method __init__ (line 9) | def __init__(self) -> None:
    method reset (line 15) | def reset(self, all_duration: str | timedelta) -> None:
    method restart_by_remain_time (line 41) | def restart_by_remain_time(self) -> None:
    method add_duration (line 50) | def add_duration(self, duration: timedelta) -> None:
    method is_timeout (line 56) | def is_timeout(self) -> bool:
    method update_remain_time (line 63) | def update_remain_time(self) -> None:
    method remain_time (line 68) | def remain_time(self) -> timedelta | None:
  class RDAgentTimerWrapper (line 75) | class RDAgentTimerWrapper(SingletonBaseClass):
    method __init__ (line 76) | def __init__(self) -> None:
    method replace_timer (line 81) | def replace_timer(self, timer: RDAgentTimer) -> None:

FILE: rdagent/log/ui/app.py
  function filter_log_folders (line 72) | def filter_log_folders(main_log_path):
  function should_display (line 131) | def should_display(msg: Message):
  function get_msgs_until (line 142) | def get_msgs_until(end_func: Callable[[Message], bool] = lambda _: True):
  function refresh (line 258) | def refresh(same_trace: bool = False):
  function evolving_feedback_window (line 292) | def evolving_feedback_window(wsf: FactorSingleFeedback | ModelSingleFeed...
  function display_hypotheses (line 327) | def display_hypotheses(hypotheses: dict[int, Hypothesis], decisions: dic...
  function metrics_window (line 370) | def metrics_window(df: pd.DataFrame, R: int, C: int, *, height: int = 30...
  function summary_window (line 424) | def summary_window():
  function tabs_hint (line 501) | def tabs_hint():
  function tasks_window (line 508) | def tasks_window(tasks: list[FactorTask | ModelTask]):
  function research_window (line 550) | def research_window():
  function feedback_window (line 586) | def feedback_window():
  function evolving_window (line 670) | def evolving_window():
  function analyze_task_completion (line 866) | def analyze_task_completion():

FILE: rdagent/log/ui/conf.py
  class UIBasePropSetting (line 6) | class UIBasePropSetting(ExtendedBaseSettings):

FILE: rdagent/log/ui/ds_summary.py
  function curves_win (line 27) | def curves_win(summary: dict):
  function all_summarize_win (line 71) | def all_summarize_win():

FILE: rdagent/log/ui/ds_trace.py
  function convert_defaultdict_to_dict (line 68) | def convert_defaultdict_to_dict(d):
  function load_data (line 74) | def load_data(log_path: Path):
  function load_stdout (line 176) | def load_stdout(stdout_path: Path):
  function task_win (line 185) | def task_win(task):
  function workspace_win (line 199) | def workspace_win(workspace, cmp_workspace=None, cmp_name="last code."):
  function show_text (line 247) | def show_text(text, lang=None):
  function highlight_prompts_uri (line 257) | def highlight_prompts_uri(uri):
  function llm_log_win (line 265) | def llm_log_win(llm_d: list):
  function hypothesis_win (line 387) | def hypothesis_win(hypo):
  function exp_gen_win (line 394) | def exp_gen_win(exp_gen_data, llm_data=None):
  function evolving_win (line 408) | def evolving_win(data, key, llm_data=None, base_workspace=None):
  function coding_win (line 452) | def coding_win(data, base_exp, llm_data: dict | None = None):
  function running_win (line 488) | def running_win(data, base_exp, llm_data=None, last_sota_exp=None):
  function feedback_win (line 524) | def feedback_win(fb_data, llm_data=None):
  function sota_win (line 540) | def sota_win(sota_exp, trace):
  function main_win (line 557) | def main_win(loop_id, llm_data=None):
  function replace_ep_path (line 609) | def replace_ep_path(p: Path):
  function get_llm_call_stats (line 621) | def get_llm_call_stats(llm_data: dict) -> tuple[int, int]:
  function get_timeout_stats (line 643) | def get_timeout_stats(llm_data: dict):
  function timedelta_to_str (line 667) | def timedelta_to_str(td: timedelta | None) -> str:
  function summarize_win (line 677) | def summarize_win():
  function stdout_win (line 1077) | def stdout_win(loop_id: int):
  function get_folders_sorted (line 1101) | def get_folders_sorted(log_path, sort_by_time=False):
  function get_state_data_range (line 1180) | def get_state_data_range(state_data):

FILE: rdagent/log/ui/ds_user_interact.py
  function render_main_content (line 21) | def render_main_content():
  function update_sessions (line 133) | def update_sessions():
  function render_sidebar (line 151) | def render_sidebar():

FILE: rdagent/log/ui/dsapp.py
  function convert_log_folder_str (line 10) | def convert_log_folder_str(lf: str) -> str:
  function extract_amlt_name (line 16) | def extract_amlt_name(x: str) -> str:

FILE: rdagent/log/ui/llm_st.py
  function get_folders_sorted (line 22) | def get_folders_sorted(log_path):
  function load_data (line 52) | def load_data():
  function show_text (line 83) | def show_text(text, lang=None):
  function highlight_prompts_uri (line 93) | def highlight_prompts_uri(uri):

FILE: rdagent/log/ui/qlib_report_figure.py
  class BaseGraph (line 9) | class BaseGraph:
    method __init__ (line 12) | def __init__(
    method _init_data (line 38) | def _init_data(self):
    method _init_parameters (line 48) | def _init_parameters(self, **kwargs):
    method get_instance_with_graph_parameters (line 62) | def get_instance_with_graph_parameters(graph_type: str = None, **kwargs):
    method _get_layout (line 77) | def _get_layout(self) -> go.Layout:
    method _get_data (line 84) | def _get_data(self) -> list:
    method figure (line 99) | def figure(self) -> go.Figure:
  class SubplotsGraph (line 110) | class SubplotsGraph:
    method __init__ (line 116) | def __init__(
    method _init_sub_graph_data (line 203) | def _init_sub_graph_data(self):
    method _init_subplots_kwargs (line 229) | def _init_subplots_kwargs(self):
    method _init_figure (line 246) | def _init_figure(self):
    method figure (line 286) | def figure(self):
  function _calculate_maximum (line 290) | def _calculate_maximum(df: pd.DataFrame, is_ex: bool = False):
  function _calculate_mdd (line 306) | def _calculate_mdd(series):
  function _calculate_report_data (line 316) | def _calculate_report_data(raw_df: pd.DataFrame) -> pd.DataFrame:
  function report_figure (line 348) | def report_figure(df: pd.DataFrame) -> list | tuple:

FILE: rdagent/log/ui/st_fixed_container.py
  function st_fixed_container (line 73) | def st_fixed_container(

FILE: rdagent/log/ui/storage.py
  class WebStorage (line 13) | class WebStorage(Storage):
    method __init__ (line 19) | def __init__(self, port: int, path: str) -> None:
    method __str__ (line 30) | def __str__(self):
    method log (line 33) | def log(self, obj: object, tag: str, timestamp: datetime | None = None...
    method truncate (line 54) | def truncate(self, time: datetime) -> None:
    method iter_msg (line 57) | def iter_msg(self, **kwargs: Any) -> Generator[Message, None, None]:
    method _obj_to_json (line 66) | def _obj_to_json(

FILE: rdagent/log/ui/utils.py
  function get_script_time (line 117) | def get_script_time(stdout_p: Path):
  function _log_path_hash_func (line 134) | def _log_path_hash_func(log_path: Path) -> str:
  function map_stat (line 146) | def map_stat(sota_mle_score: dict | None) -> str:
  function get_best_report (line 164) | def get_best_report(log_path: Path) -> dict | None:
  function _get_sota_exp_stat_hash_func (line 182) | def _get_sota_exp_stat_hash_func(log_path: Path, selector: Literal["auto...
  function get_sota_exp_stat (line 186) | def get_sota_exp_stat(
  function _get_score_stat_hash_func (line 262) | def _get_score_stat_hash_func(log_path: Path, sota_loop_id: int) -> str:
  function get_score_stat (line 266) | def get_score_stat(log_path: Path, sota_loop_id: int) -> tuple[float | N...
  function load_times_deprecated (line 365) | def load_times_deprecated(log_path: Path):
  function load_times_info (line 382) | def load_times_info(log_path: Path) -> dict[int, dict[str, dict[Literal[...
  function _log_folders_summary_hash_func (line 423) | def _log_folders_summary_hash_func(log_folder: str | Path, hours: int | ...
  function get_summary_df (line 432) | def get_summary_df(log_folder: str | Path, hours: int | None = None) -> ...
  function percent_df (line 668) | def percent_df(summary_df: pd.DataFrame, show_origin=True) -> pd.DataFrame:
  function get_statistics_df (line 712) | def get_statistics_df(summary_df: pd.DataFrame) -> pd.DataFrame:
  function curve_figure (line 783) | def curve_figure(scores: pd.DataFrame) -> go.Figure:
  function lite_curve_figure (line 818) | def lite_curve_figure(summary):
  function trace_figure (line 884) | def trace_figure(trace: Trace, merge_loops: list = []):
  function timeline_figure (line 975) | def timeline_figure(times_dict: dict[int, dict[str, dict[Literal["start_...
  function compare (line 1089) | def compare(

FILE: rdagent/log/ui/web.py
  class WebView (line 29) | class WebView(View):
    method __init__ (line 30) | def __init__(self, ui: "StWindow"):
    method display (line 35) | def display(self, s: Storage, watch: bool = False):
  class StWindow (line 42) | class StWindow:
    method __init__ (line 43) | def __init__(self, container: "DeltaGenerator"):
    method consume_msg (line 46) | def consume_msg(self, msg: Message):
  class LLMWindow (line 51) | class LLMWindow(StWindow):
    method __init__ (line 52) | def __init__(self, container: "DeltaGenerator", session_name: str = "c...
    method consume_msg (line 56) | def consume_msg(self, msg: Message):
  class ProgressTabsWindow (line 60) | class ProgressTabsWindow(StWindow):
    method __init__ (line 65) | def __init__(
    method consume_msg (line 78) | def consume_msg(self, msg: Message):
  class ObjectsTabsWindow (line 102) | class ObjectsTabsWindow(StWindow):
    method __init__ (line 103) | def __init__(
    method consume_msg (line 115) | def consume_msg(self, msg: Message):
  class RoundTabsWindow (line 145) | class RoundTabsWindow(StWindow):
    method __init__ (line 146) | def __init__(
    method consume_msg (line 161) | def consume_msg(self, msg: Message):
  class HypothesisWindow (line 169) | class HypothesisWindow(StWindow):
    method consume_msg (line 170) | def consume_msg(self, msg: Message | Hypothesis):
  class HypothesisFeedbackWindow (line 179) | class HypothesisFeedbackWindow(StWindow):
    method consume_msg (line 180) | def consume_msg(self, msg: Message | HypothesisFeedback):
  class FactorTaskWindow (line 192) | class FactorTaskWindow(StWindow):
    method consume_msg (line 193) | def consume_msg(self, msg: Message | FactorTask):
  class ModelTaskWindow (line 206) | class ModelTaskWindow(StWindow):
    method consume_msg (line 207) | def consume_msg(self, msg: Message | ModelTask):
  class FactorFeedbackWindow (line 220) | class FactorFeedbackWindow(StWindow):
    method consume_msg (line 221) | def consume_msg(self, msg: Message | FactorSingleFeedback):
  class ModelFeedbackWindow (line 237) | class ModelFeedbackWindow(StWindow):
    method consume_msg (line 238) | def consume_msg(self, msg: Message | ModelSingleFeedback):
  class WorkspaceWindow (line 256) | class WorkspaceWindow(StWindow):
    method __init__ (line 257) | def __init__(self, container: "DeltaGenerator", show_task_info: bool =...
    method consume_msg (line 261) | def consume_msg(self, msg: Message | FactorFBWorkspace | ModelFBWorksp...
  class QlibFactorExpWindow (line 285) | class QlibFactorExpWindow(StWindow):
    method __init__ (line 286) | def __init__(self, container: DeltaGenerator, show_task_info: bool = F...
    method consume_msg (line 290) | def consume_msg(self, msg: Message | QlibFactorExperiment):
  class QlibModelExpWindow (line 318) | class QlibModelExpWindow(StWindow):
    method __init__ (line 319) | def __init__(self, container: DeltaGenerator, show_task_info: bool = F...
    method consume_msg (line 323) | def consume_msg(self, msg: Message | QlibModelExperiment):
  class SimpleTraceWindow (line 345) | class SimpleTraceWindow(StWindow):
    method __init__ (line 346) | def __init__(
    method consume_msg (line 358) | def consume_msg(self, msg: Message):
  function mock_msg (line 433) | def mock_msg(obj) -> Message:
  class TraceObjWindow (line 437) | class TraceObjWindow(StWindow):
    method __init__ (line 438) | def __init__(self, container: "DeltaGenerator" = st.container()):
    method consume_msg (line 441) | def consume_msg(self, msg: Message | Trace):
  class ResearchWindow (line 457) | class ResearchWindow(StWindow):
    method consume_msg (line 458) | def consume_msg(self, msg: Message):
  class EvolvingWindow (line 477) | class EvolvingWindow(StWindow):
    method __init__ (line 478) | def __init__(self, container: "DeltaGenerator"):
    method consume_msg (line 482) | def consume_msg(self, msg: Message):
  class DevelopmentWindow (line 519) | class DevelopmentWindow(StWindow):
    method __init__ (line 520) | def __init__(self, container: "DeltaGenerator"):
    method consume_msg (line 528) | def consume_msg(self, msg: Message):
  class FeedbackWindow (line 533) | class FeedbackWindow(StWindow):
    method __init__ (line 534) | def __init__(self, container: "DeltaGenerator"):
    method consume_msg (line 537) | def consume_msg(self, msg: Message):
  class SingleRDLoopWindow (line 550) | class SingleRDLoopWindow(StWindow):
    method __init__ (line 551) | def __init__(self, container: "DeltaGenerator"):
    method consume_msg (line 558) | def consume_msg(self, msg: Message):
  class TraceWindow (line 568) | class TraceWindow(StWindow):
    method __init__ (line 569) | def __init__(
    method consume_msg (line 598) | def consume_msg(self, msg: Message):

FILE: rdagent/log/utils/__init__.py
  class LogColors (line 9) | class LogColors:
    method get_all_colors (line 30) | def get_all_colors(cls: type["LogColors"]) -> list:
    method render (line 35) | def render(self, text: str, color: str = "", style: str = "") -> str:
    method remove_ansi_codes (line 55) | def remove_ansi_codes(s: str) -> str:
  class CallerInfo (line 63) | class CallerInfo(TypedDict):
  function get_caller_info (line 69) | def get_caller_info(level: int = 2) -> CallerInfo:
  function is_valid_session (line 83) | def is_valid_session(log_path: Path) -> bool:
  function extract_loopid_func_name (line 87) | def extract_loopid_func_name(tag: str) -> tuple[str, str] | tuple[None, ...
  function extract_evoid (line 93) | def extract_evoid(tag: str) -> str | None:
  function extract_json (line 99) | def extract_json(log_content: str) -> dict | None:
  function gen_datetime (line 106) | def gen_datetime(dt: datetime | None = None) -> datetime:
  function dict_get_with_warning (line 117) | def dict_get_with_warning(d: dict, key: str, default: Any = None) -> Any:

FILE: rdagent/log/utils/folder.py
  function get_first_session_file_after_duration (line 14) | def get_first_session_file_after_duration(log_folder: str | Path, durati...
  function first_li_si_after_one_time (line 40) | def first_li_si_after_one_time(log_path: Path, hours: int = 12) -> tuple...

FILE: rdagent/oai/backend/base.py
  class JSONParser (line 37) | class JSONParser:
    method __init__ (line 40) | def __init__(self, add_json_in_prompt: bool = False) -> None:
    method parse (line 49) | def parse(self, content: str) -> str:
    method _direct_parse (line 71) | def _direct_parse(self, content: str) -> str:
    method _extract_from_code_block (line 81) | def _extract_from_code_block(self, content: str) -> str:
    method _fix_python_syntax (line 90) | def _fix_python_syntax(self, content: str) -> str:
    method _extract_with_fix_combined (line 95) | def _extract_with_fix_combined(self, content: str) -> str:
    method _fix_python_booleans (line 107) | def _fix_python_booleans(json_str: str) -> str:
    method _extract_first_json (line 132) | def _extract_first_json(response: str) -> str:
  class CodeBlockParser (line 139) | class CodeBlockParser:
    method __init__ (line 150) | def __init__(self, language: str = "python", fallback_to_raw: bool = F...
    method _get_language_aliases (line 161) | def _get_language_aliases(self, language: str) -> List[str]:
    method parse (line 168) | def parse(self, content: str) -> str:
  class SQliteLazyCache (line 195) | class SQliteLazyCache(SingletonBaseClass):
    method __init__ (line 196) | def __init__(self, cache_location: str) -> None:
    method chat_get (line 230) | def chat_get(self, key: str) -> str | None:
    method embedding_get (line 236) | def embedding_get(self, key: str) -> list | dict | str | None:
    method chat_set (line 242) | def chat_set(self, key: str, value: str) -> None:
    method embedding_set (line 251) | def embedding_set(self, content_to_embedding_dict: dict) -> None:
    method message_get (line 260) | def message_get(self, conversation_id: str) -> list[dict[str, Any]]:
    method message_set (line 265) | def message_set(self, conversation_id: str, message_value: list[dict[s...
  class SessionChatHistoryCache (line 274) | class SessionChatHistoryCache(SingletonBaseClass):
    method __init__ (line 275) | def __init__(self) -> None:
    method message_get (line 279) | def message_get(self, conversation_id: str) -> list[dict[str, Any]]:
    method message_set (line 282) | def message_set(self, conversation_id: str, message_value: list[dict[s...
  class ChatSession (line 286) | class ChatSession:
    method __init__ (line 287) | def __init__(self, api_backend: Any, conversation_id: str | None = Non...
    method build_chat_completion_message (line 292) | def build_chat_completion_message(self, user_prompt: str) -> list[dict...
    method build_chat_completion_message_and_calculate_token (line 305) | def build_chat_completion_message_and_calculate_token(self, user_promp...
    method build_chat_completion (line 309) | def build_chat_completion(self, user_prompt: str, *args, **kwargs) -> ...
    method get_conversation_id (line 345) | def get_conversation_id(self) -> str:
    method display_history (line 348) | def display_history(self) -> None:
  class APIBackend (line 353) | class APIBackend(ABC):
    method __init__ (line 360) | def __init__(
    method build_chat_session (line 381) | def build_chat_session(
    method _build_messages (line 392) | def _build_messages(
    method _build_log_messages (line 429) | def _build_log_messages(self, messages: list[dict[str, Any]]) -> str:
    method build_messages_and_create_chat_completion (line 440) | def build_messages_and_create_chat_completion(  # type: ignore[no-unty...
    method create_embedding (line 493) | def create_embedding(self, input_content: str | list[str], *args, **kw...
    method build_messages_and_calculate_token (line 505) | def build_messages_and_calculate_token(
    method _try_create_chat_completion_or_embedding (line 520) | def _try_create_chat_completion_or_embedding(  # type: ignore[no-untyp...
    method _add_json_in_prompt (line 615) | def _add_json_in_prompt(self, messages: list[dict[str, Any]]) -> None:
    method _create_chat_completion_auto_continue (line 625) | def _create_chat_completion_auto_continue(
    method _create_embedding_with_cache (line 733) | def _create_embedding_with_cache(
    method supports_response_schema (line 757) | def supports_response_schema(self) -> bool:
    method _calculate_token_from_messages (line 764) | def _calculate_token_from_messages(self, messages: list[dict[str, Any]...
    method _create_embedding_inner_function (line 771) | def _create_embedding_inner_function(self, input_content_list: list[st...
    method _create_chat_completion_inner_function (line 778) | def _create_chat_completion_inner_function(  # type: ignore[no-untyped...
    method chat_token_limit (line 791) | def chat_token_limit(self) -> int:

FILE: rdagent/oai/backend/deprec.py
  class ConvManager (line 64) | class ConvManager:
    method __init__ (line 70) | def __init__(
    method _rotate_files (line 79) | def _rotate_files(self) -> None:
    method append (line 92) | def append(self, conv: tuple[list, str]) -> None:
  class DeprecBackend (line 99) | class DeprecBackend(APIBackend):
    method __init__ (line 109) | def __init__(  # noqa: C901, PLR0912, PLR0915
    method _get_encoder (line 238) | def _get_encoder(self) -> tiktoken.Encoding:
    method supports_response_schema (line 266) | def supports_response_schema(self) -> bool:
    method _create_embedding_inner_function (line 273) | def _create_embedding_inner_function(self, input_content_list: list[st...
    method _create_chat_completion_inner_function (line 294) | def _create_chat_completion_inner_function(  # type: ignore[no-untyped...
    method _calculate_token_from_messages (line 467) | def _calculate_token_from_messages(self, messages: list[dict[str, Any]...

FILE: rdagent/oai/backend/litellm.py
  function _reduce_no_init (line 25) | def _reduce_no_init(exc: Exception) -> tuple:
  class LiteLLMSettings (line 35) | class LiteLLMSettings(LLMSettings):
    class Config (line 37) | class Config:
  class LiteLLMAPIBackend (line 48) | class LiteLLMAPIBackend(APIBackend):
    method __init__ (line 53) | def __init__(self, *args: Any, **kwargs: Any) -> None:
    method _calculate_token_from_messages (line 60) | def _calculate_token_from_messages(self, messages: list[dict[str, Any]...
    method _create_embedding_inner_function (line 71) | def _create_embedding_inner_function(self, input_content_list: list[st...
    class CompleteKwargs (line 89) | class CompleteKwargs(TypedDict):
    method get_complete_kwargs (line 95) | def get_complete_kwargs(self) -> CompleteKwargs:
    method _create_chat_completion_inner_function (line 127) | def _create_chat_completion_inner_function(  # type: ignore[no-untyped...
    method supports_response_schema (line 226) | def supports_response_schema(self) -> bool:
    method chat_token_limit (line 233) | def chat_token_limit(self) -> int:

FILE: rdagent/oai/backend/pydantic_ai.py
  function get_agent_model (line 27) | def get_agent_model() -> OpenAIChatModel:

FILE: rdagent/oai/llm_conf.py
  class LLMSettings (line 11) | class LLMSettings(ExtendedBaseSettings):

FILE: rdagent/oai/llm_utils.py
  function calculate_embedding_distance_between_str_list (line 13) | def calculate_embedding_distance_between_str_list(
  function get_api_backend (line 35) | def get_api_backend(*args: Any, **kwargs: Any) -> BaseAPIBackend:  # TOD...

FILE: rdagent/oai/utils/embedding.py
  function get_embedding_max_tokens (line 27) | def get_embedding_max_tokens(model: str) -> int:
  function trim_text_for_embedding (line 63) | def trim_text_for_embedding(text: str, model: str, max_tokens: Optional[...
  function truncate_content_list (line 118) | def truncate_content_list(content_list: list[str], model: str) -> list[s...

FILE: rdagent/scenarios/data_science/debug/data.py
  class DataHandler (line 18) | class DataHandler:
    method load (line 21) | def load(self, path) -> pd.DataFrame:
    method dump (line 24) | def dump(self, df: pd.DataFrame, path):
  class GenericDataHandler (line 28) | class GenericDataHandler(DataHandler):
    method load (line 34) | def load(self, path) -> pd.DataFrame:
    method dump (line 65) | def dump(self, df: pd.DataFrame | dict, path):
  class DataReducer (line 93) | class DataReducer:
    method __init__ (line 96) | def __init__(self, min_frac=0.02, min_num=5):
    method reduce (line 101) | def reduce(self, df: pd.DataFrame) -> pd.DataFrame:
  class RandDataReducer (line 105) | class RandDataReducer(DataReducer):
    method reduce (line 111) | def reduce(self, df: pd.DataFrame, frac: float = None) -> pd.DataFrame:
  class FolderReducer (line 119) | class FolderReducer(DataReducer):
    method reduce (line 124) | def reduce(self, array: list, frac: float = None) -> list:
  class FileReducer (line 141) | class FileReducer(DataReducer):
    method reduce (line 146) | def reduce(self, files: list[Path]) -> list:
  class FileKeepReducer (line 158) | class FileKeepReducer(DataReducer):
    method reduce (line 163) | def reduce(self, files: list[Path]) -> list:
  class SingleFileReducer (line 181) | class SingleFileReducer(DataReducer):
    method reduce (line 186) | def reduce(self, files: list[Path]) -> list:
  class UniqueIDDataReducer (line 191) | class UniqueIDDataReducer(DataReducer):
    method reduce (line 192) | def reduce(self, df: pd.DataFrame) -> pd.DataFrame:
  class JsonReducer (line 237) | class JsonReducer(DataReducer):
    method extract_filename (line 239) | def extract_filename(self, item: Any) -> Optional[str]:
    method reduce (line 255) | def reduce(self, data: dict) -> dict:
    method _find_all_lists (line 274) | def _find_all_lists(
    method _sample_list (line 299) | def _sample_list(self, lst: List[Any]) -> List[Any]:
  class DataSampler (line 306) | class DataSampler:
    method __init__ (line 309) | def __init__(self, data_folder, sample_folder, reducer):
    method sample (line 316) | def sample(self) -> None:
  class DefaultSampler (line 320) | class DefaultSampler(DataSampler):
    method sample (line 321) | def sample(self) -> None:
  class FolderSampler (line 467) | class FolderSampler(DataSampler):
    method sample (line 472) | def sample(self) -> None:
  class SingleFilePerFolderSampler (line 520) | class SingleFilePerFolderSampler(DataSampler):
    method sample (line 526) | def sample(self) -> None:
  function copy_file (line 558) | def copy_file(src_fp, target_folder, data_folder):
  function copy_folder (line 569) | def copy_folder(src_fp, target_folder, data_folder):
  function count_files_in_folder (line 579) | def count_files_in_folder(files_to_process):
  function map_competition (line 592) | def map_competition(competition: str) -> tuple[DataReducer, DataSampler]:
  function create_debug_data (line 605) | def create_debug_data(

FILE: rdagent/scenarios/data_science/dev/feedback.py
  class DSExperiment2Feedback (line 23) | class DSExperiment2Feedback(Experiment2Feedback):
    method __init__ (line 24) | def __init__(self, scen: Scenario, version: str = "exp_feedback") -> N...
    method generate_feedback (line 28) | def generate_feedback(self, exp: DSExperiment, trace: DSTrace) -> Expe...

FILE: rdagent/scenarios/data_science/dev/runner/__init__.py
  class DSRunnerCoSTEERSettings (line 29) | class DSRunnerCoSTEERSettings(CoSTEERSettings):
    class Config (line 32) | class Config:
  class DSRunnerMultiProcessEvolvingStrategy (line 42) | class DSRunnerMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 44) | def implement_one_task(
    method assign_code_list_to_evo (line 116) | def assign_code_list_to_evo(self, code_list: list[dict[str, str]], evo):
  class DSCoSTEERRunner (line 135) | class DSCoSTEERRunner(CoSTEER):
    method __init__ (line 136) | def __init__(
    method get_develop_max_seconds (line 169) | def get_develop_max_seconds(self) -> int | None:
    method should_use_new_evo (line 175) | def should_use_new_evo(self, base_fb: CoSTEERMultiFeedback | None, new...
    method develop (line 196) | def develop(self, exp):

FILE: rdagent/scenarios/data_science/dev/runner/eval.py
  class DSRunnerFeedback (line 34) | class DSRunnerFeedback(CoSTEERSingleFeedback):
    method is_acceptable (line 45) | def is_acceptable(self) -> bool:
    method __str__ (line 50) | def __str__(self) -> str:
  class DSRunnerEvaluator (line 72) | class DSRunnerEvaluator(CoSTEEREvaluator):
    method evaluate (line 74) | def evaluate(

FILE: rdagent/scenarios/data_science/example/arf-12-hours-prediction-task/sample.py
  function sample_and_copy_subfolder (line 10) | def sample_and_copy_subfolder(
  function copy_other_file (line 46) | def copy_other_file(source: Path, target: Path):
  function create_debug_data (line 63) | def create_debug_data(

FILE: rdagent/scenarios/data_science/example/eval/arf-12-hours-prediction-task/grade.py
  function prepare_for_auroc_metric (line 7) | def prepare_for_auroc_metric(submission: pd.DataFrame, answers: pd.DataF...
  function grade (line 46) | def grade(submission: pd.DataFrame, answers: pd.DataFrame) -> float:

FILE: rdagent/scenarios/data_science/example/eval/playground-series-s4e9/grade.py
  class InvalidSubmissionError (line 9) | class InvalidSubmissionError(Exception):
  function prepare_for_metric (line 17) | def prepare_for_metric(submission: pd.DataFrame, answers: pd.DataFrame) ...
  function grade (line 42) | def grade(submission: pd.DataFrame, answers: pd.DataFrame) -> float:

FILE: rdagent/scenarios/data_science/example/source_data/playground-series-s4e9/prepare.py
  function prepare (line 7) | def prepare(raw: Path, public: Path, private: Path):

FILE: rdagent/scenarios/data_science/experiment/experiment.py
  class DSExperiment (line 11) | class DSExperiment(Experiment[Task, FBWorkspace, FBWorkspace]):
    method __init__ (line 12) | def __init__(self, pending_tasks_list: list, hypothesis_candidates: li...
    method set_user_instructions (line 27) | def set_user_instructions(self, user_instructions: UserInstructions | ...
    method is_ready_to_run (line 35) | def is_ready_to_run(self) -> bool:
    method set_local_selection (line 42) | def set_local_selection(self, local_selection: tuple[int, ...]) -> None:

FILE: rdagent/scenarios/data_science/interactor/__init__.py
  class DSInteractor (line 17) | class DSInteractor(Interactor[DSExperiment]):
    method dump_and_wait_for_user_input (line 19) | def dump_and_wait_for_user_input(
    method interact (line 32) | def interact(self, exp: DSExperiment, trace: DSTrace) -> DSExperiment:
  class FBDSInteractor (line 70) | class FBDSInteractor(DSInteractor):
    method dump_and_wait_for_user_input (line 71) | def dump_and_wait_for_user_input(

FILE: rdagent/scenarios/data_science/loop.py
  function clean_workspace (line 43) | def clean_workspace(workspace_root: Path) -> None:
  function backup_folder (line 64) | def backup_folder(path: str | Path) -> Path:
  class DataScienceRDLoop (line 87) | class DataScienceRDLoop(RDLoop):
    method __init__ (line 96) | def __init__(self, PROP_SETTING: BasePropSetting):
    method direct_exp_gen (line 139) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 154) | def coding(self, prev_out: dict[str, Any]):
    method running (line 177) | def running(self, prev_out: dict[str, Any]):
    method feedback (line 187) | def feedback(self, prev_out: dict[str, Any]) -> ExperimentFeedback:
    method record (line 211) | def record(self, prev_out: dict[str, Any]):
    method _check_exit_conditions_on_step (line 353) | def _check_exit_conditions_on_step(self, loop_id: Optional[int] = None...
    method load (line 359) | def load(
    method dump (line 373) | def dump(self, path: str | Path) -> None:

FILE: rdagent/scenarios/data_science/proposal/exp_gen/base.py
  class DSHypothesis (line 13) | class DSHypothesis(Hypothesis):
    method __init__ (line 14) | def __init__(
    method __str__ (line 37) | def __str__(self) -> str:
  class DSTrace (line 55) | class DSTrace(Trace[DataScienceScen, KnowledgeBase]):
    method __init__ (line 56) | def __init__(self, scen: DataScienceScen, knowledge_base: KnowledgeBas...
    method should_inject_diversity (line 66) | def should_inject_diversity(self, current_selection: tuple[int, ...] |...
    method register_uncommitted_exp (line 81) | def register_uncommitted_exp(self, exp: DSExperiment, loop_id: int):
    method deregister_uncommitted_exp (line 84) | def deregister_uncommitted_exp(self, loop_id: int):
    method set_sota_exp_to_submit (line 88) | def set_sota_exp_to_submit(self, exp: DSExperiment) -> None:
    method sub_trace_count (line 92) | def sub_trace_count(self) -> int:
    method get_leaves (line 95) | def get_leaves(self) -> list[int, ...]:
    method get_sibling_exps (line 114) | def get_sibling_exps(self, current_selection: tuple[int, ...] | None =...
    method sync_dag_parent_and_hist (line 139) | def sync_dag_parent_and_hist(
    method retrieve_search_list (line 165) | def retrieve_search_list(
    method next_incomplete_component (line 194) | def next_incomplete_component(
    method has_component (line 212) | def has_component(
    method experiment_and_feedback_list_after_init (line 221) | def experiment_and_feedback_list_after_init(
    method sota_experiment_fb (line 267) | def sota_experiment_fb(
    method sota_experiment (line 287) | def sota_experiment(
    method last_successful_exp (line 297) | def last_successful_exp(
    method last_exp (line 312) | def last_exp(
    method last_exp_fb (line 323) | def last_exp_fb(
    method last_runnable_exp_fb (line 336) | def last_runnable_exp_fb(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/diversity_strategy.py
  class DiversityContextStrategy (line 10) | class DiversityContextStrategy(ABC):
    method should_inject (line 17) | def should_inject(self, trace: DSTrace, local_selection: tuple[int, .....
  class InjectAtRootStrategy (line 32) | class InjectAtRootStrategy(DiversityContextStrategy):
    method should_inject (line 37) | def should_inject(self, trace: DSTrace, local_selection: tuple[int, .....
  class InjectUntilSOTAGainedStrategy (line 42) | class InjectUntilSOTAGainedStrategy(DiversityContextStrategy):
    method should_inject (line 48) | def should_inject(self, trace: DSTrace, local_selection: tuple[int, .....
  class AlwaysInjectStrategy (line 61) | class AlwaysInjectStrategy(DiversityContextStrategy):
    method should_inject (line 66) | def should_inject(self, trace: DSTrace, local_selection: tuple[int, .....

FILE: rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py
  class DSDraftExpGen (line 26) | class DSDraftExpGen(ExpGen):
    method _init_task_gen (line 27) | def _init_task_gen(
    method gen (line 61) | def gen(
  class DSDraftV2ExpGen (line 132) | class DSDraftV2ExpGen(ExpGen):
    method __init__ (line 133) | def __init__(self, *args, **kwargs):
    method tag_gen (line 137) | def tag_gen(self, scenario_desc: str) -> str:
    method knowledge_gen (line 150) | def knowledge_gen(self) -> str:
    method hypothesis_gen (line 157) | def hypothesis_gen(
    method task_gen (line 183) | def task_gen(
    method gen (line 241) | def gen(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/idea_pool.py
  class DSIdea (line 18) | class DSIdea:
    method __init__ (line 19) | def __init__(self, raw_knowledge: Dict | str) -> None:
    method __str__ (line 41) | def __str__(self) -> str:
    method to_formatted_str (line 52) | def to_formatted_str(self) -> str:
  class DSKnowledgeBase (line 56) | class DSKnowledgeBase(UndirectedGraph):
    method __init__ (line 57) | def __init__(self, path: str | Path | None = None, idea_pool_json_path...
    method add_idea (line 64) | def add_idea(self, idea: List[DSIdea] | DSIdea) -> None:
    method build_idea_pool (line 97) | def build_idea_pool(self, idea_pool_json_path: str | Path):
    method sample_ideas (line 116) | def sample_ideas(
    method update_pickled_problem (line 182) | def update_pickled_problem(self, problems: Dict, pickled_problem_name:...

FILE: rdagent/scenarios/data_science/proposal/exp_gen/merge.py
  class MergeExpGen (line 26) | class MergeExpGen(ExpGen):
    method gen (line 27) | def gen(
  class ExpGen2Hypothesis (line 99) | class ExpGen2Hypothesis(DSProposalV2ExpGen):
    method hypothesis_gen (line 101) | def hypothesis_gen(
    method get_exp_index (line 131) | def get_exp_index(self, trace: DSTrace) -> int:
    method gen (line 148) | def gen(
  class ExpGen2TraceAndMerge (line 239) | class ExpGen2TraceAndMerge(ExpGen):
    method __init__ (line 241) | def __init__(self, *args, **kwargs):
    method gen (line 246) | def gen(
  class MergeExpGen_MultiTrace (line 275) | class MergeExpGen_MultiTrace(ExpGen):
    method gen (line 276) | def gen(
  class ExpGen2TraceAndMergeV2 (line 354) | class ExpGen2TraceAndMergeV2(ExpGen):
    method __init__ (line 355) | def __init__(self, *args, **kwargs):
    method reset_exp_gen_version (line 361) | def reset_exp_gen_version(self, version: str = "v2"):
    method gen (line 370) | def gen(
  class ExpGen2TraceAndMergeV3 (line 414) | class ExpGen2TraceAndMergeV3(ExpGen):
    method __init__ (line 415) | def __init__(self, *args, **kwargs):
    method gen (line 420) | def gen(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/naive.py
  class NaiveExpGen (line 14) | class NaiveExpGen(ExpGen):
    method gen (line 15) | def gen(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/package_info.py
  function get_installed_packages (line 5) | def get_installed_packages():
  function print_filtered_packages (line 9) | def print_filtered_packages(installed_packages, filtered_packages):
  function get_python_packages (line 24) | def get_python_packages():

FILE: rdagent/scenarios/data_science/proposal/exp_gen/planner/__init__.py
  class DSExperimentPlan (line 9) | class DSExperimentPlan(ExperimentPlan):
    method __init__ (line 15) | def __init__(self):
  class DSExpPlannerHandCraft (line 22) | class DSExpPlannerHandCraft(ExpPlanner[DSExperimentPlan]):
    method plan (line 27) | def plan(self, trace: DSTrace) -> DSExperimentPlan:

FILE: rdagent/scenarios/data_science/proposal/exp_gen/proposal.py
  function get_component (line 85) | def get_component(name: str) -> Dict[str, Any]:
  class ScenarioChallengeCategory (line 98) | class ScenarioChallengeCategory(str, Enum):
  class ScenarioChallengeDetail (line 103) | class ScenarioChallengeDetail(BaseModel):
  class ScenarioAnalysis (line 121) | class ScenarioAnalysis(BaseModel):
  class ScenarioChallenges (line 134) | class ScenarioChallenges(BaseModel):
  class TraceAnalysisDetail (line 145) | class TraceAnalysisDetail(BaseModel):
  class TraceAnalysis (line 155) | class TraceAnalysis(BaseModel):
  class TraceChallengeDetail (line 168) | class TraceChallengeDetail(BaseModel):
  class TraceChallenges (line 195) | class TraceChallenges(BaseModel):
  class HypothesisComponent (line 210) | class HypothesisComponent(str, Enum):
  class HypothesisEvaluationReasoningScore (line 218) | class HypothesisEvaluationReasoningScore(BaseModel):
  class HypothesisEvaluation (line 225) | class HypothesisEvaluation(BaseModel):
  class HypothesisDetail (line 243) | class HypothesisDetail(BaseModel):
  class HypothesisSimple (line 260) | class HypothesisSimple(BaseModel):
  class HypothesisList (line 267) | class HypothesisList(BaseModel):
  class CodingSketch (line 276) | class CodingSketch(BaseModel):
  function draft_exp_in_decomposition (line 306) | def draft_exp_in_decomposition(scen: Scenario, trace: DSTrace) -> None |...
  class DSProposalV1ExpGen (line 317) | class DSProposalV1ExpGen(ExpGen):
    method gen (line 318) | def gen(
  class DSProposalV2ExpGen (line 492) | class DSProposalV2ExpGen(ExpGen):
    method __init__ (line 493) | def __init__(self, *args, **kwargs):
    method identify_scenario_problem (line 497) | def identify_scenario_problem(
    method identify_feedback_problem (line 532) | def identify_feedback_problem(
    method identify_problem (line 569) | def identify_problem(
    method hypothesis_gen (line 609) | def hypothesis_gen(
    method hypothesis_critique (line 705) | def hypothesis_critique(
    method hypothesis_rewrite (line 774) | def hypothesis_rewrite(
    method compute_top_scores (line 859) | def compute_top_scores(
    method select_hypothesis (line 893) | def select_hypothesis(
    method _cosine_similarity_matrix_torch (line 924) | def _cosine_similarity_matrix_torch(self, A, B):
    method _prob_dis_torch (line 932) | def _prob_dis_torch(
    method _get_path (line 987) | def _get_path(self, node, parent_nodes):
    method _get_current_exp_score_list (line 995) | def _get_current_exp_score_list(self, trace, competition):
    method _llm_select_extra_hypo (line 1028) | def _llm_select_extra_hypo(self, trace: DSTrace) -> list[tuple[str, fl...
    method hypothesis_select_with_llm (line 1055) | def hypothesis_select_with_llm(
    method hypothesis_rank (line 1158) | def hypothesis_rank(
    method task_gen (line 1184) | def task_gen(
    method get_all_hypotheses (line 1283) | def get_all_hypotheses(self, problem_dict: dict, hypothesis_dict: dict...
    method gen (line 1300) | def gen(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/router/__init__.py
  class ParallelMultiTraceExpGen (line 36) | class ParallelMultiTraceExpGen(ExpGen):
    method __init__ (line 45) | def __init__(self, *args, **kwargs):
    method gen (line 58) | def gen(
    method reset (line 67) | def reset(self) -> None:
    method async_gen (line 70) | async def async_gen(self, trace: DSTrace, loop: LoopBase) -> DSExperim...

FILE: rdagent/scenarios/data_science/proposal/exp_gen/select/expand.py
  class LatestCKPSelector (line 13) | class LatestCKPSelector(CheckpointSelector):
    method __init__ (line 18) | def __init__(
    method get_selection (line 23) | def get_selection(self, trace: Trace) -> tuple[int, ...]:
  class LimitTimeCKPSelector (line 28) | class LimitTimeCKPSelector(CheckpointSelector):
    method __init__ (line 33) | def __init__(
    method set_time_limit (line 41) | def set_time_limit(self):
    method get_selection (line 52) | def get_selection(self, trace: Trace) -> tuple[int, ...]:
  class SOTAJumpCKPSelector (line 100) | class SOTAJumpCKPSelector(CheckpointSelector):
    method __init__ (line 107) | def __init__(
    method get_selection (line 118) | def get_selection(self, trace: Trace) -> tuple[int, ...]:
  class BackJumpCKPSelector (line 156) | class BackJumpCKPSelector(CheckpointSelector):
    method __init__ (line 164) | def __init__(
    method get_selection (line 175) | def get_selection(self, trace: Trace) -> tuple[int, ...]:

FILE: rdagent/scenarios/data_science/proposal/exp_gen/select/submit.py
  class GlobalSOTASelector (line 43) | class GlobalSOTASelector(SOTAexpSelector):
    method __init__ (line 48) | def __init__(self):
    method get_sota_exp_to_submit (line 51) | def get_sota_exp_to_submit(self, trace: Trace, **kwargs) -> DSExperime...
  class AutoSOTAexpSelector (line 58) | class AutoSOTAexpSelector(SOTAexpSelector):
    method __init__ (line 64) | def __init__(self):
    method get_sota_exp_to_submit (line 68) | def get_sota_exp_to_submit(self, trace: Trace, **kwargs) -> DSExperime...
    method collect_sota_candidates (line 130) | def collect_sota_candidates(self, trace: Trace) -> list:
  class BestValidSelector (line 158) | class BestValidSelector(SOTAexpSelector):
    method __init__ (line 164) | def __init__(self, num_candidates: int = 1, use_decision: bool = True,...
    method get_sota_exp_to_submit (line 178) | def get_sota_exp_to_submit(self, trace: Trace, **kwargs) -> DSExperime...
    method collect_sota_candidates (line 187) | def collect_sota_candidates(self, trace: Trace) -> list[DSExperiment] ...
  class ValidationSelector (line 263) | class ValidationSelector(SOTAexpSelector):
    method __init__ (line 271) | def __init__(
    method get_sota_exp_to_submit (line 289) | def get_sota_exp_to_submit(self, trace: Trace) -> DSExperiment | None:
    method print_code (line 345) | def print_code(self, data_py_code: str, grade_py_code: str):
    method _prepare_validation_scripts (line 353) | def _prepare_validation_scripts(
    method _generate_and_run_script (line 429) | def _generate_and_run_script(
  function process_experiment (line 521) | def process_experiment(
  function _parsing_score (line 575) | def _parsing_score(grade_stdout: str) -> Optional[float]:
  function check_hit (line 602) | def check_hit(selected_exp: DSExperiment, trace: Trace, sota_result: Dic...
  function try_get_loop_id (line 622) | def try_get_loop_id(trace: Trace, exp: DSExperiment):
  function extract_tar (line 629) | def extract_tar(tar_path: str, to_dir: str = "log") -> str:
  function evaluate_one_trace (line 639) | def evaluate_one_trace(
  function select_on_existing_trace (line 744) | def select_on_existing_trace(

FILE: rdagent/scenarios/data_science/proposal/exp_gen/trace_scheduler.py
  class TraceScheduler (line 18) | class TraceScheduler(ABC):
    method next (line 25) | async def next(self, trace: DSTrace) -> tuple[int, ...]:
    method reset (line 43) | def reset(self) -> None:
  class BaseScheduler (line 50) | class BaseScheduler(TraceScheduler):
    method __init__ (line 51) | def __init__(self):
    method next (line 55) | async def next(self, trace: DSTrace) -> tuple[int, ...]:
    method process_uncommitted_nodes (line 85) | def process_uncommitted_nodes(self, trace: DSTrace) -> None:
    method select (line 93) | def select(self, trace: DSTrace) -> tuple[int, ...] | None:
    method reset (line 97) | def reset(self) -> None:
  class RoundRobinScheduler (line 102) | class RoundRobinScheduler(BaseScheduler):
    method __init__ (line 110) | def __init__(self, max_trace_num: int, *args, **kwargs):
    method select (line 116) | def select(self, trace: DSTrace) -> tuple[int, ...] | None:
  class ProbabilisticScheduler (line 139) | class ProbabilisticScheduler(BaseScheduler):
    method __init__ (line 145) | def __init__(self, max_trace_num: int, temperature: float = 1.0, *args...
    method calculate_potential (line 160) | def calculate_potential(self, trace: DSTrace, leaf_id: int) -> float:
    method _softmax_probabilities (line 174) | def _softmax_probabilities(self, potentials: list[float]) -> list[float]:
    method select (line 201) | def select(self, trace: DSTrace) -> tuple[int, ...] | None:
  class TraceLengthScheduler (line 232) | class TraceLengthScheduler(ProbabilisticScheduler):
    method __init__ (line 240) | def __init__(self, max_trace_num: int, temperature: float = 1.0, inver...
    method calculate_potential (line 253) | def calculate_potential(self, trace: DSTrace, leaf_id: int) -> float:
  class SOTABasedScheduler (line 267) | class SOTABasedScheduler(ProbabilisticScheduler):
    method __init__ (line 272) | def __init__(self, max_trace_num: int, temperature: float = 1.0, inver...
    method calculate_potential (line 283) | def calculate_potential(self, trace: DSTrace, leaf_id: int) -> float:
  class RandomScheduler (line 304) | class RandomScheduler(ProbabilisticScheduler):
    method calculate_potential (line 309) | def calculate_potential(self, trace: DSTrace, leaf_id: int) -> float:
  class MCTSScheduler (line 316) | class MCTSScheduler(ProbabilisticScheduler):
    method __init__ (line 331) | def __init__(self, max_trace_num: int, temperature: float = 1.0, *args...
    method _get_q (line 344) | def _get_q(self, node_id: int) -> float:
    method _get_u (line 352) | def _get_u(self, node_id: int) -> float:
    method select (line 358) | def select(self, trace: DSTrace) -> tuple[int, ...] | None:
    method observe_feedback (line 396) | def observe_feedback(self, trace: DSTrace, new_idx: int) -> None:
    method reset (line 420) | def reset(self) -> None:
    method process_uncommitted_nodes (line 431) | def process_uncommitted_nodes(self, trace: DSTrace) -> None:

FILE: rdagent/scenarios/data_science/proposal/exp_gen/utils.py
  function get_component (line 56) | def get_component(name: str) -> Dict[str, Any]:
  class CodingSketch (line 69) | class CodingSketch(BaseModel):
  function get_packages (line 94) | def get_packages(pkgs: list[str] | None = None) -> str:

FILE: rdagent/scenarios/data_science/scen/__init__.py
  class DataScienceScen (line 27) | class DataScienceScen(Scenario):
    method __init__ (line 30) | def __init__(self, competition: str) -> None:
    method reanalyze_competition_description (line 70) | def reanalyze_competition_description(self):
    method _get_description (line 76) | def _get_description(self):
    method _get_direction (line 89) | def _get_direction(self):
    method _analysis_competition_description (line 92) | def _analysis_competition_description(self):
    method real_debug_timeout (line 141) | def real_debug_timeout(self):
    method recommend_debug_timeout (line 152) | def recommend_debug_timeout(self):
    method real_full_timeout (line 155) | def real_full_timeout(self):
    method recommend_full_timeout (line 176) | def recommend_full_timeout(self):
    method increase_timeout (line 179) | def increase_timeout(self):
    method background (line 184) | def background(self) -> str:
    method rich_style_description (line 197) | def rich_style_description(self) -> str:
    method get_competition_full_desc (line 203) | def get_competition_full_desc(self) -> str:
    method get_scenario_all_desc (line 220) | def get_scenario_all_desc(self, eda_output=None) -> str:
    method get_runtime_environment (line 246) | def get_runtime_environment(self) -> str:
    method _get_data_folder_description (line 253) | def _get_data_folder_description(self) -> str:
  class KaggleScen (line 259) | class KaggleScen(DataScienceScen):
    method __init__ (line 268) | def __init__(self, competition: str) -> None:
    method _get_description (line 272) | def _get_description(self):
    method _get_direction (line 275) | def _get_direction(self):
    method rich_style_description (line 279) | def rich_style_description(self) -> str:

FILE: rdagent/scenarios/data_science/scen/utils.py
  class FileTreeGenerationError (line 91) | class FileTreeGenerationError(Exception):
  class MaxLinesExceededError (line 97) | class MaxLinesExceededError(FileTreeGenerationError):
  class DirectoryPermissionError (line 103) | class DirectoryPermissionError(FileTreeGenerationError):
  function get_file_len_size (line 109) | def get_file_len_size(f: Path) -> Tuple[int, str]:
  function preview_df (line 122) | def preview_df(df: pd.DataFrame, file_name: str, simple=True, show_nan_c...
  function preview_csv (line 177) | def preview_csv(p: Path, file_name: str, simple=True, show_nan_columns=F...
  function preview_parquet (line 183) | def preview_parquet(p: Path, file_name: str, simple=True, show_nan_colum...
  function preview_json (line 189) | def preview_json(p: Path, file_name: str):
  function _walk (line 272) | def _walk(path: Path):
  class FileTreeGenerator (line 292) | class FileTreeGenerator:
    method __init__ (line 297) | def __init__(
    method _build_lookup_set (line 322) | def _build_lookup_set(self):
    method generate_tree (line 338) | def generate_tree(self, path: Union[str, Path]) -> str:
    method _add_line (line 409) | def _add_line(self, text: str) -> None:
    method _process_directory (line 424) | def _process_directory(self, path: Path, depth: int, prefix: str, base...
    method _process_subdirectories (line 471) | def _process_subdirectories(self, dirs: List[Path], depth: int, prefix...
    method _process_single_directory (line 491) | def _process_single_directory(self, d: Path, depth: int, prefix: str, ...
    method _process_files (line 513) | def _process_files(self, all_files: List[Path], depth: int, prefix: st...
    method _categorize_files (line 533) | def _categorize_files(self, files: List[Path]) -> Tuple[List[Path], Li...
    method _get_size_str (line 549) | def _get_size_str(self, file_path: Path) -> str:
  class DataFolderDescriptor (line 558) | class DataFolderDescriptor:
    method __init__ (line 563) | def __init__(self, tree_generator: FileTreeGenerator = None):
    method describe_folder (line 572) | def describe_folder(
    method _select_files_for_preview (line 647) | def _select_files_for_preview(
  function file_tree_v2 (line 689) | def file_tree_v2(path: Union[str, Path], max_lines: int = 200, priority_...
  function describe_data_folder_v2 (line 695) | def describe_data_folder_v2(

FILE: rdagent/scenarios/data_science/test_eval.py
  class NoTestEvalError (line 9) | class NoTestEvalError(Exception):
  class TestEvalBase (line 13) | class TestEvalBase:
    method eval (line 17) | def eval(self, competition: str, workspace: FBWorkspace) -> str:
    method valid (line 21) | def valid(self, competition: str, workspace: FBWorkspace) -> tuple[str...
    method enabled (line 25) | def enabled(self, competition) -> bool:
    method get_sample_submission_name (line 29) | def get_sample_submission_name(self, competition: str) -> str:
    method is_sub_enabled (line 47) | def is_sub_enabled(self, competition: str) -> bool:
  class TestEval (line 62) | class TestEval(TestEvalBase):
    method __init__ (line 65) | def __init__(self) -> None:
    method eval (line 69) | def eval(self, competition: str, workspace: FBWorkspace) -> str:
    method valid (line 84) | def valid(self, competition: str, workspace: FBWorkspace) -> tuple[str...
    method enabled (line 101) | def enabled(self, competition) -> bool:
  class MLETestEval (line 107) | class MLETestEval(TestEvalBase):
    method __init__ (line 110) | def __init__(self) -> None:
    method eval (line 117) | def eval(self, competition: str, workspace: FBWorkspace) -> str:
    method valid (line 126) | def valid(self, competition: str, workspace: FBWorkspace) -> tuple[str...
    method enabled (line 138) | def enabled(self, competition) -> bool:
  function get_test_eval (line 142) | def get_test_eval() -> TestEvalBase:

FILE: rdagent/scenarios/finetune/benchmark/benchmark.py
  function get_model_inference_config (line 49) | def get_model_inference_config(base_model_name: str, gpu_count: int) -> ...
  function detect_model_type (line 94) | def detect_model_type(model_path: str) -> bool:
  function run_benchmark (line 115) | def run_benchmark(
  function get_benchmark_ranges (line 331) | def get_benchmark_ranges() -> tuple[str, str]:

FILE: rdagent/scenarios/finetune/benchmark/data/adaptor.py
  class BenchmarkConfig (line 20) | class BenchmarkConfig:

FILE: rdagent/scenarios/finetune/benchmark/data/default.py
  function _to_bool (line 22) | def _to_bool(value: Any) -> bool:
  function _is_correct (line 38) | def _is_correct(sample: Dict) -> bool:
  function _format_value (line 62) | def _format_value(value: Any) -> str:
  function _format_prompt (line 71) | def _format_prompt(prompt: Any) -> str:
  function _extract_tag_content (line 111) | def _extract_tag_content(prompt: Any, tag_name: str) -> str:
  function _get_question (line 137) | def _get_question(sample: Dict, pred_entry: Dict) -> str:
  function _get_gold (line 160) | def _get_gold(sample: Dict, pred_entry: Dict) -> str:
  function _get_prediction (line 182) | def _get_prediction(sample: Dict, pred_entry: Dict) -> str:
  function extract_error_samples (line 208) | def extract_error_samples(

FILE: rdagent/scenarios/finetune/benchmark/data/financeiq_gen.py
  function download_financeiq_dataset (line 15) | def download_financeiq_dataset() -> None:
  function extract_error_samples (line 55) | def extract_error_samples(results_base: Path, max_samples: int = 10) -> ...

FILE: rdagent/scenarios/finetune/benchmark/merge/merge.py
  function is_blackwell_gpu (line 12) | def is_blackwell_gpu() -> bool:
  function check_if_merging_needed (line 29) | def check_if_merging_needed(model_path: str | Path) -> bool:
  function merge_model (line 50) | def merge_model(env, workspace_path: Path, base_model_path: str, adapter...

FILE: rdagent/scenarios/finetune/datasets/__init__.py
  class DatasetConfig (line 23) | class DatasetConfig:
  function _remove_eval_splits (line 35) | def _remove_eval_splits(out_dir: str) -> None:
  function prepare (line 71) | def prepare(name: str, force: bool = False) -> str:
  function prepare_all (line 113) | def prepare_all(force: bool = False) -> dict[str, str]:

FILE: rdagent/scenarios/finetune/datasets/chemcot/__init__.py
  function normalize_rcr (line 7) | def normalize_rcr(out_dir: str) -> None:

FILE: rdagent/scenarios/finetune/datasets/financeiq/split.py
  function get_split_indices (line 7) | def get_split_indices(
  function split_financeiq_dataset (line 26) | def split_financeiq_dataset(data_dir: str, split: Literal["train", "test...

FILE: rdagent/scenarios/finetune/dev/feedback.py
  class FTExperiment2Feedback (line 28) | class FTExperiment2Feedback(Experiment2Feedback):
    method __init__ (line 31) | def __init__(self, scen: Scenario, version: str = "exp_feedback") -> N...
    method generate_feedback (line 35) | def generate_feedback(

FILE: rdagent/scenarios/finetune/download/hf.py
  function _ensure_parent (line 7) | def _ensure_parent(path: Path) -> None:
  function _get_hf_token (line 11) | def _get_hf_token(token: Optional[str] = None) -> Optional[str]:
  function download_dataset (line 21) | def download_dataset(
  function download_model (line 66) | def download_model(

FILE: rdagent/scenarios/finetune/experiment/experiment.py
  class FTExperiment (line 13) | class FTExperiment(Experiment[Task, FTWorkspace, FTWorkspace]):
    method __init__ (line 14) | def __init__(self, sub_tasks: list[Task], *args, **kwargs) -> None:
    method is_ready_to_run (line 27) | def is_ready_to_run(self) -> bool:

FILE: rdagent/scenarios/finetune/experiment/workspace.py
  class FTWorkspace (line 27) | class FTWorkspace(FBWorkspace):
    method __init__ (line 37) | def __init__(self, *args, **kwargs):
    method run (line 48) | def run(

FILE: rdagent/scenarios/finetune/loop.py
  class LLMFinetuneRDLoop (line 14) | class LLMFinetuneRDLoop(RDLoop):
    method __init__ (line 20) | def __init__(self, PROP_SETTING: LLMFinetunePropSetting):
    method direct_exp_gen (line 32) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 39) | def coding(self, prev_out: dict[str, Any]):
    method feedback (line 46) | def feedback(self, prev_out: dict[str, Any]):
    method record (line 57) | def record(self, prev_out: dict[str, Any]):

FILE: rdagent/scenarios/finetune/proposal/proposal.py
  class FTHypothesis (line 24) | class FTHypothesis(Hypothesis):
    method __init__ (line 27) | def __init__(
    method __str__ (line 43) | def __str__(self) -> str:
  class LLMFinetuneExpGen (line 56) | class LLMFinetuneExpGen(ExpGen):
    method __init__ (line 62) | def __init__(self, scen: LLMFinetuneScen):
    method gen (line 65) | def gen(self, trace: Trace) -> FTExperiment:
    method _gen_hypothesis (line 74) | def _gen_hypothesis(self, trace: Trace, base_model: str, parent_exp: F...

FILE: rdagent/scenarios/finetune/proposal/trace.py
  class FTTrace (line 23) | class FTTrace(Trace["LLMFinetuneScen", KnowledgeBase]):
    method __init__ (line 30) | def __init__(self, scen: "LLMFinetuneScen", knowledge_base: KnowledgeB...
    method sota_benchmark (line 36) | def sota_benchmark(self) -> dict | None:
    method get_experiment_info (line 49) | def get_experiment_info(self, exp: "FTExperiment") -> dict[str, Any]:
    method sota_info (line 75) | def sota_info(self) -> dict[str, Any] | None:

FILE: rdagent/scenarios/finetune/scen/docker_scripts/extract_parameters.py
  function extract_field_info (line 31) | def extract_field_info(field):
  function extract_params (line 51) | def extract_params(cls):
  function extract_base_params (line 56) | def extract_base_params(cls):
  function save_parameters (line 73) | def save_parameters(base_dir):
  function main (line 109) | def main():

FILE: rdagent/scenarios/finetune/scen/llama_factory_manager.py
  class LLaMAFactoryManager (line 99) | class LLaMAFactoryManager:
    method __init__ (line 102) | def __init__(self):
    method extract_info_from_docker (line 107) | def extract_info_from_docker(self) -> Dict:
    method _load_extracted_data (line 154) | def _load_extracted_data(self) -> Dict:
    method get_info (line 172) | def get_info(self) -> Dict:
    method methods (line 179) | def methods(self) -> List[str]:
    method models (line 184) | def models(self) -> List[str]:
    method hf_models (line 189) | def hf_models(self) -> List[str]:
    method peft_methods (line 195) | def peft_methods(self) -> List[str]:
    method training_stages (line 201) | def training_stages(self) -> Dict[str, str]:
    method templates (line 206) | def templates(self) -> List[str]:
    method is_peft_method (line 210) | def is_peft_method(self, method: str) -> bool:
    method get_parameters (line 214) | def get_parameters(self, param_type: Optional[str] = None) -> Dict:
    method _format_param_line (line 221) | def _format_param_line(self, param_name: str, param_info: dict, max_he...
    method _format_params_dict (line 238) | def _format_params_dict(self, params_dict: dict, max_help_len: int | N...
    method format_shared_params (line 246) | def format_shared_params(self, max_help_len: int | None = DEFAULT_HELP...
    method format_method_specific_params (line 263) | def format_method_specific_params(self, method: str, max_help_len: int...

FILE: rdagent/scenarios/finetune/scen/memory_estimator.py
  class MemoryEstimator (line 10) | class MemoryEstimator:
    method __init__ (line 32) | def __init__(
    method from_model_name (line 52) | def from_model_name(
    method _base_memory (line 73) | def _base_memory(self, method: str) -> float:
    method _activation_factor (line 87) | def _activation_factor(self, method: str) -> float:
    method _find_max_seq_len (line 91) | def _find_max_seq_len(self, method: str, batch_size: int = 1) -> int:
    method estimate (line 105) | def estimate(self) -> dict[str, int]:
    method format (line 110) | def format(self, estimates: dict[str, int] = None) -> str:

FILE: rdagent/scenarios/finetune/scen/scenario.py
  class LLMFinetuneScen (line 26) | class LLMFinetuneScen(DataScienceScen):
    method __init__ (line 29) | def __init__(self) -> None:
    method benchmark_hash (line 74) | def benchmark_hash(self, model_name, benchmark_name) -> str:
    method run_baseline_model_evaluation (line 78) | def run_baseline_model_evaluation(self, model_name, benchmark_name) ->...
    method real_full_timeout (line 112) | def real_full_timeout(self):
    method _generate_memory_report (line 115) | def _generate_memory_report(self) -> str:
    method _validate_and_prepare_environment (line 150) | def _validate_and_prepare_environment(self):
    method _initialize_llama_factory (line 164) | def _initialize_llama_factory(self):
    method _select_relevant_datasets (line 175) | def _select_relevant_datasets(self) -> list[str]:
    method _llm_select_datasets (line 201) | def _llm_select_datasets(self) -> tuple[list[str], str]:
    method _prepare_dataset_config (line 231) | def _prepare_dataset_config(self) -> dict:
    method metric_direction (line 280) | def metric_direction(self) -> bool:
    method get_scenario_all_desc (line 284) | def get_scenario_all_desc(self, enable_dataset_description: bool = Fal...

FILE: rdagent/scenarios/finetune/scen/utils.py
  function _find_data_files (line 21) | def _find_data_files(dataset_path: Path, max_files: int = 50) -> list[Pa...
  function _truncate_long_values (line 40) | def _truncate_long_values(obj, max_length: int = 3000):
  function _compute_column_stats (line 67) | def _compute_column_stats(data: list[dict]) -> dict[str, dict]:
  function _load_dataset_for_stats (line 138) | def _load_dataset_for_stats(data_files: list[Path], max_samples: int = 5...
  class FinetuneDatasetDescription (line 187) | class FinetuneDatasetDescription(dict):
    method __str__ (line 190) | def __str__(self) -> str:
  class FinetuneFileDescription (line 216) | class FinetuneFileDescription(dict):
    method __str__ (line 219) | def __str__(self) -> str:
  class FinetuneDatasetDescriptor (line 230) | class FinetuneDatasetDescriptor:
    method _generate_file_tree (line 233) | def _generate_file_tree(self, dataset_path: Path) -> str:
    method _count_samples_in_file (line 242) | def _count_samples_in_file(self, data_file: Path) -> int:
    method _generate_stats (line 275) | def _generate_stats(self, dataset_path: Path, include_column_stats: bo...
    method hash_dataset_path (line 334) | def hash_dataset_path(
    method describe_dataset_folder (line 348) | def describe_dataset_folder(
    method get_dataset_stats (line 425) | def get_dataset_stats(self, dataset_path: Path) -> dict[str, Any]:
    method _walk (line 429) | def _walk(self, dir_path: Path, depth: int, max_depth: int, target_nam...
    method _read_dataset_readme (line 445) | def _read_dataset_readme(self, dataset_path: Path, max_chars: int = 50...
    method _extract_samples_for_template (line 467) | def _extract_samples_for_template(self, data_files: list[Path], max_sa...
    method describe_model (line 502) | def describe_model(self, base_model_name: str = None, ft_file_path: st...
    method describe_file_json (line 579) | def describe_file_json(self, data_file: Path, max_samples: int = 3) ->...
    method describe_file_jsonl (line 594) | def describe_file_jsonl(self, data_file: Path, max_samples: int = 3) -...
    method describe_file_csv (line 616) | def describe_file_csv(self, data_file: Path, max_samples: int = 3) -> ...
    method describe_file_parquet (line 634) | def describe_file_parquet(self, data_file: Path, max_samples: int = 3)...
    method describe_data_file (line 652) | def describe_data_file(self, data_file: Path) -> FinetuneFileDescription:
    method _discover_subtasks (line 677) | def _discover_subtasks(self, dataset_dir: Path) -> dict:
    method analyze_dataset (line 712) | def analyze_dataset(self, dataset_dir: Path) -> dict:
  function _read_single_dataset_readme (line 774) | def _read_single_dataset_readme(dataset_path: Path, max_chars: int = 200...
  function check_all_dataset_in_info (line 830) | def check_all_dataset_in_info(ft_file_path, existing_config, max_depth: ...
  function generate_dataset_info_config (line 863) | def generate_dataset_info_config(target_dataset_list: list, ft_file_path...

FILE: rdagent/scenarios/finetune/train/eval.py
  function extract_loss_history (line 30) | def extract_loss_history(output_path) -> Dict[str, List[Dict[str, Any]]]:
  class FTRunnerEvaluator (line 78) | class FTRunnerEvaluator(CoSTEEREvaluator):
    method evaluate (line 81) | def evaluate(
    method _generate_llm_feedback (line 239) | def _generate_llm_feedback(
    method _run_full_data_processing (line 303) | def _run_full_data_processing(self, implementation: FBWorkspace):

FILE: rdagent/scenarios/finetune/train/runner.py
  class FTRunnerSettings (line 31) | class FTRunnerSettings(FTCoderCoSTEERSettings):
    class Config (line 34) | class Config:
  class FTRunnerEvolvingStrategy (line 38) | class FTRunnerEvolvingStrategy(MultiProcessEvolvingStrategy):
    method implement_one_task (line 45) | def implement_one_task(
  class LLMFinetuneRunner (line 63) | class LLMFinetuneRunner(CoSTEER):
    method __init__ (line 66) | def __init__(
    method develop (line 95) | def develop(self, exp):
    method get_develop_max_seconds (line 110) | def get_develop_max_seconds(self) -> int | None:
    method compare_and_pick_fb (line 114) | def compare_and_pick_fb(self, base_fb, new_fb) -> bool:

FILE: rdagent/scenarios/finetune/utils.py
  function ensure_ft_assets_exist (line 9) | def ensure_ft_assets_exist(

FILE: rdagent/scenarios/general_model/scenario.py
  class GeneralModelScenario (line 8) | class GeneralModelScenario(Scenario):
    method __init__ (line 9) | def __init__(self) -> None:
    method background (line 18) | def background(self) -> str:
    method source_data (line 22) | def source_data(self) -> str:
    method output_format (line 26) | def output_format(self) -> str:
    method interface (line 30) | def interface(self) -> str:
    method simulator (line 34) | def simulator(self) -> str:
    method rich_style_description (line 38) | def rich_style_description(self) -> str:
    method get_scenario_all_desc (line 41) | def get_scenario_all_desc(
    method get_runtime_environment (line 54) | def get_runtime_environment(self):

FILE: rdagent/scenarios/kaggle/developer/coder.py
  class KGModelFeatureSelectionCoder (line 35) | class KGModelFeatureSelectionCoder(Developer[KGModelExperiment]):
    method develop (line 36) | def develop(self, exp: KGModelExperiment) -> KGModelExperiment:

FILE: rdagent/scenarios/kaggle/developer/feedback.py
  class KGExperiment2Feedback (line 16) | class KGExperiment2Feedback(Experiment2Feedback):
    method process_results (line 17) | def process_results(self, current_result, sota_result):
    method generate_feedback (line 40) | def generate_feedback(self, exp: Experiment, trace: Trace) -> Hypothes...

FILE: rdagent/scenarios/kaggle/developer/runner.py
  class KGCachedRunner (line 17) | class KGCachedRunner(CachedRunner[ASpecificExp]):
    method get_cache_key (line 18) | def get_cache_key(self, exp: ASpecificExp) -> str:
    method assign_cached_result (line 28) | def assign_cached_result(self, exp: Experiment, cached_res: Experiment...
    method init_develop (line 41) | def init_develop(self, exp: KGFactorExperiment | KGModelExperiment) ->...
  class KGModelRunner (line 60) | class KGModelRunner(KGCachedRunner[KGModelExperiment]):
    method develop (line 62) | def develop(self, exp: KGModelExperiment) -> KGModelExperiment:
  class KGFactorRunner (line 94) | class KGFactorRunner(KGCachedRunner[KGFactorExperiment]):
    method develop (line 96) | def develop(self, exp: KGFactorExperiment) -> KGFactorExperiment:

FILE: rdagent/scenarios/kaggle/experiment/kaggle_experiment.py
  class KGModelExperiment (line 37) | class KGModelExperiment(ModelExperiment[ModelTask, KGFBWorkspace, ModelF...
    method __init__ (line 38) | def __init__(self, *args, source_feature_size: int = None, **kwargs) -...
  class KGFactorExperiment (line 61) | class KGFactorExperiment(FeatureExperiment[FactorTask, KGFBWorkspace, Fa...
    method __init__ (line 62) | def __init__(self, *args, source_feature_size: int = None, **kwargs) -...

FILE: rdagent/scenarios/kaggle/experiment/scenario.py
  class KGScenario (line 36) | class KGScenario(Scenario):
    method __init__ (line 37) | def __init__(self, competition: str) -> None:
    method _analysis_competition_description (line 73) | def _analysis_competition_description(self):
    method get_competition_full_desc (line 101) | def get_competition_full_desc(self) -> str:
    method background (line 114) | def background(self) -> str:
    method source_data (line 134) | def source_data(self) -> str:
    method output_format (line 167) | def output_format(self, tag=None) -> str:
    method interface (line 182) | def interface(self, tag=None) -> str:
    method simulator (line 197) | def simulator(self, tag=None) -> str:
    method rich_style_description (line 215) | def rich_style_description(self) -> str:
    method get_scenario_all_desc (line 242) | def get_scenario_all_desc(

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 31) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 58) | def preprocess_transform(X: pd.DataFrame, preprocessor, label_encoders):
  function preprocess_script (line 77) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 27) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/spaceship-titanic_template/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 24) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/covid19-global-forecasting-week-1/fea_share_preprocess.py
  function prepreprocess (line 9) | def prepreprocess():
  function preprocess_script (line 45) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/covid19-global-forecasting-week-1/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/covid19-global-forecasting-week-1/model/model_xgboost.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 27) | def predict(models, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/covid19-global-forecasting-week-1/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/covid19-global-forecasting-week-1/train.py
  function compute_rmsle (line 17) | def compute_rmsle(y_true, y_pred):
  function import_module_from_path (line 22) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/fea_share_preprocess.py
  function prepreprocess (line 9) | def prepreprocess():
  function preprocess_script (line 26) | def preprocess_script():
  function clean_and_impute_data (line 54) | def clean_and_impute_data(X_train, X_valid, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/model_nn.py
  class NeuralNetwork (line 12) | class NeuralNetwork(nn.Module):
    method __init__ (line 13) | def __init__(self, input_channels, num_classes):
    method forward (line 23) | def forward(self, x):
  function fit (line 34) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 82) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/model_xgboost.py
  function fit (line 9) | def fit(X_train, y_train, X_valid, y_valid):
  function predict (line 30) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/digit-recognizer/train.py
  function compute_metrics_for_classification (line 17) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 22) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/fea_share_preprocess.py
  function preprocess_script (line 9) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/feature/feature.py
  class IdentityFeature (line 10) | class IdentityFeature:
    method fit (line 11) | def fit(self, train_df: pd.DataFrame):
    method transform (line 18) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/model/model_randomforest.py
  function fit (line 6) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 19) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/model/model_xgboost.py
  function is_sparse_df (line 10) | def is_sparse_df(df: pd.DataFrame) -> bool:
  function fit (line 15) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 30) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/feedback-prize-english-language-learning/train.py
  function import_module_from_path (line 11) | def import_module_from_path(module_name, module_path):
  function MCRMSE (line 18) | def MCRMSE(y_true, y_pred):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/fea_share_preprocess.py
  function prepreprocess (line 9) | def prepreprocess():
  function preprocess_script (line 26) | def preprocess_script():
  function clean_and_impute_data (line 50) | def clean_and_impute_data(X_train, X_valid, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 31) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 29) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/train.py
  function compute_metrics_for_classification (line 17) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 23) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/forest-cover-type-prediction/train_past.py
  function import_module_from_path (line 19) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 31) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 63) | def preprocess_transform(X: pd.DataFrame, preprocessor):
  function preprocess_script (line 83) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/model/model_nn.py
  class FeatureInteractionModel (line 13) | class FeatureInteractionModel(nn.Module):
    method __init__ (line 14) | def __init__(self, num_features):
    method forward (line 23) | def forward(self, x):
  function fit (line 32) | def fit(X_train, y_train, X_valid, y_valid):
  function predict (line 67) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/model/model_randomforest.py
  function select (line 13) | def select(X: pd.DataFrame) -> pd.DataFrame:
  function fit (line 21) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 43) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/model/model_xgboost.py
  function select (line 9) | def select(X: pd.DataFrame) -> pd.DataFrame:
  function fit (line 14) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 34) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/meta_tpl_deprecated/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function compute_metrics_for_classification (line 24) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 30) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/fea_share_preprocess.py
  function prepreprocess (line 12) | def prepreprocess():
  function preprocess_script (line 29) | def preprocess_script():
  function clean_and_impute_data (line 53) | def clean_and_impute_data(X_train, X_valid, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/feature/feature.py
  class DatetimeFeature (line 9) | class DatetimeFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/model_linear.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 31) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/select_linear.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/new-york-city-taxi-fare-prediction/train.py
  function compute_metrics_for_classification (line 17) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 23) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/fea_share_preprocess.py
  function prepreprocess (line 12) | def prepreprocess():
  function preprocess_fit (line 37) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 62) | def preprocess_transform(X: pd.DataFrame, preprocessor, numerical_cols, ...
  function preprocess_script (line 70) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/model_randomforest.py
  function fit (line 7) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/model_xgboost.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 25) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/optiver-realized-volatility-prediction/train.py
  function compute_rmspe (line 17) | def compute_rmspe(y_true, y_pred):
  function import_module_from_path (line 23) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/fea_share_preprocess.py
  function preprocess_script (line 8) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/model_randomforest.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 22) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 29) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e11/train.py
  function import_module_from_path (line 12) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/fea_share_preprocess.py
  function preprocess_script (line 8) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/model_randomforest.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 22) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 29) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e14/train.py
  function import_module_from_path (line 12) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/fea_share_preprocess.py
  function preprocess_script (line 9) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/model_randomforest.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 22) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 29) | def predict(model, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e16/train.py
  function import_module_from_path (line 12) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 48) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 67) | def preprocess_transform(X: pd.DataFrame, preprocessor):
  function preprocess_script (line 80) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/model_xgboost.py
  function fit (line 10) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 32) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s3e26/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 25) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/fea_share_preprocess.py
  function prepreprocess (line 10) | def prepreprocess():
  function preprocess_fit (line 22) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 38) | def preprocess_transform(X: pd.DataFrame, preprocessor, numerical_cols):
  function preprocess_script (line 47) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/model_randomforest.py
  function fit (line 7) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/model_xgboost.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 28) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e5/train.py
  function compute_r2 (line 17) | def compute_r2(y_true, y_pred):
  function import_module_from_path (line 22) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 31) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 62) | def preprocess_transform(X: pd.DataFrame, preprocessor, numerical_cols, ...
  function preprocess_script (line 71) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 31) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 27) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e8/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 24) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 23) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 48) | def preprocess_transform(X: pd.DataFrame, preprocessor, numerical_cols, ...
  function preprocess_script (line 57) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/model_randomforest.py
  function fit (line 7) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/model_xgboost.py
  function fit (line 5) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 25) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/playground-series-s4e9/train.py
  function compute_rmse (line 18) | def compute_rmse(y_true, y_pred):
  function import_module_from_path (line 25) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 69) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 88) | def preprocess_transform(X: pd.DataFrame, preprocessor):
  function preprocess_script (line 101) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/model_randomforest.py
  function fit (line 12) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 25) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/model_xgboost.py
  function fit (line 10) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 32) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/sf-crime/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 25) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/fea_share_preprocess.py
  function prepreprocess (line 11) | def prepreprocess():
  function preprocess_fit (line 31) | def preprocess_fit(X_train: pd.DataFrame):
  function preprocess_transform (line 58) | def preprocess_transform(X: pd.DataFrame, preprocessor, label_encoders):
  function preprocess_script (line 77) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 26) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 27) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/spaceship-titanic/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 24) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/fea_share_preprocess.py
  function prepreprocess (line 8) | def prepreprocess():
  function preprocess_script (line 53) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/model/model_xgboost.py
  function fit (line 10) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 32) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/statoil-iceberg-classifier-challenge/train.py
  function compute_metrics_for_classification (line 18) | def compute_metrics_for_classification(y_true, y_pred):
  function import_module_from_path (line 23) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/fea_share_preprocess.py
  function prepreprocess (line 12) | def prepreprocess():
  function preprocess_script (line 29) | def preprocess_script():
  function clean_and_impute_data (line 62) | def clean_and_impute_data(X_train, X_valid, X_test):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 31) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 30) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-dec-2021/train.py
  function import_module_from_path (line 17) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/fea_share_preprocess.py
  function preprocess_script (line 8) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 31) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 29) | def predict(model: xgb.Booster, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/tabular-playground-series-may-2022/train.py
  function import_module_from_path (line 17) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/fea_share_preprocess.py
  function preprocess_script (line 8) | def preprocess_script():

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/feature/feature.py
  class IdentityFeature (line 9) | class IdentityFeature:
    method fit (line 10) | def fit(self, train_df: pd.DataFrame):
    method transform (line 16) | def transform(self, X: pd.DataFrame):

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/model_randomforest.py
  function fit (line 13) | def fit(X_train: pd.DataFrame, y_train: pd.Series, X_valid: pd.DataFrame...
  function predict (line 33) | def predict(model, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/model_xgboost.py
  function fit (line 9) | def fit(X_train: pd.DataFrame, y_train: pd.DataFrame, X_valid: pd.DataFr...
  function predict (line 34) | def predict(model: xgb.Booster, X):

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/select_lightgbm.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/select_nn.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/select_randomforest.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/model/select_xgboost.py
  function select (line 4) | def select(X: pd.DataFrame) -> pd.DataFrame:

FILE: rdagent/scenarios/kaggle/experiment/templates/ventilator-pressure-prediction/train.py
  function import_module_from_path (line 17) | def import_module_from_path(module_name, module_path):

FILE: rdagent/scenarios/kaggle/experiment/utils.py
  function python_files_to_notebook (line 6) | def python_files_to_notebook(competition: str, py_dir: str):

FILE: rdagent/scenarios/kaggle/experiment/workspace.py
  class KGFBWorkspace (line 28) | class KGFBWorkspace(FBWorkspace):
    method __init__ (line 29) | def __init__(self, template_folder_path: Path, *args, **kwargs) -> None:
    method model_description (line 35) | def model_description(self) -> dict[str, str]:
    method generate_preprocess_data (line 42) | def generate_preprocess_data(
    method execute (line 72) | def execute(self, run_env: dict = {}, *args, **kwargs) -> str:

FILE: rdagent/scenarios/kaggle/kaggle_crawler.py
  function crawl_descriptions (line 35) | def crawl_descriptions(
  function download_data (line 110) | def download_data(competition: str, settings: ExtendedBaseSettings, enab...
  function unzip_data (line 203) | def unzip_data(unzip_file_path: str, unzip_target_path: str) -> None:
  function leaderboard_scores (line 209) | def leaderboard_scores(competition: str) -> list[float]:
  function get_metric_direction (line 218) | def get_metric_direction(competition: str) -> bool:
  function score_rank (line 232) | def score_rank(competition: str, score: float) -> tuple[int, float]:
  function download_notebooks (line 252) | def download_notebooks(competition: str, local_path: str, num: int = 15)...
  function notebook_to_knowledge (line 275) | def notebook_to_knowledge(notebook_text: str) -> str:
  function convert_notebooks_to_text (line 287) | def convert_notebooks_to_text(competition: str, local_path: str) -> None:
  function collect_knowledge_texts (line 323) | def collect_knowledge_texts(notebooks_path: str | Path) -> dict[str, lis...

FILE: rdagent/scenarios/kaggle/knowledge_management/extract_knowledge.py
  function extract_knowledge_from_high_score_answers (line 8) | def extract_knowledge_from_high_score_answers(content: str):
  function extract_knowledge_from_feedback (line 26) | def extract_knowledge_from_feedback(feedback_response: dict) -> dict:
  function process_all_case_files (line 49) | def process_all_case_files(directory_path: str):

FILE: rdagent/scenarios/kaggle/knowledge_management/graph.py
  class KGKnowledgeGraph (line 20) | class KGKnowledgeGraph(UndirectedGraph):
    method __init__ (line 21) | def __init__(self, path: str | Path | None, scenario: KGScenario | Non...
    method add_document (line 37) | def add_document(self, document_content: str, scenario: KGScenario | N...
    method analyze_one_document (line 41) | def analyze_one_document(self, document_content: str, scenario: KGScen...
    method load_from_documents (line 60) | def load_from_documents(self, documents: List[str], scenario: KGScenar...

FILE: rdagent/scenarios/kaggle/knowledge_management/vector_base.py
  class KGKnowledgeDocument (line 16) | class KGKnowledgeDocument(Document):
    method __init__ (line 21) | def __init__(
    method split_into_trunk (line 66) | def split_into_trunk(self, size: int = 1000, overlap: int = 0):
    method from_dict (line 82) | def from_dict(self, data: dict):
    method __repr__ (line 96) | def __repr__(self):
  class KaggleExperienceBase (line 106) | class KaggleExperienceBase(PDVectorBase):
    method __init__ (line 111) | def __init__(self, vector_df_path: Union[str, Path] = None, kaggle_exp...
    method add (line 128) | def add(self, document: Union[KGDocument, List[KGDocument]]):
    method load_kaggle_experience (line 162) | def load_kaggle_experience(self, kaggle_experience_path: Union[str, Pa...
    method add_experience_to_vector_base (line 179) | def add_experience_to_vector_base(self, experiment_feedback=None):
    method search_experience (line 228) | def search_experience(self, target: str, query: str, topk_k: int = 5, ...
    method refine_with_LLM (line 270) | def refine_with_LLM(self, target: str, text: str) -> str:
    method save (line 282) | def save(self, vector_df_path: Union[str, Path]):

FILE: rdagent/scenarios/kaggle/proposal/proposal.py
  class KGHypothesis (line 31) | class KGHypothesis(Hypothesis):
    method __init__ (line 32) | def __init__(
    method __str__ (line 47) | def __str__(self) -> str:
  function generate_RAG_content (line 58) | def generate_RAG_content(
  class KGHypothesisGen (line 185) | class KGHypothesisGen(FactorAndModelHypothesisGen):
    method __init__ (line 197) | def __init__(self, scen: Scenario) -> Tuple[dict, bool]:
    method update_reward_estimates (line 200) | def update_reward_estimates(self, trace: Trace) -> None:
    method execute_next_action (line 228) | def execute_next_action(self, trace: Trace) -> str:
    method prepare_context (line 250) | def prepare_context(self, trace: Trace) -> Tuple[dict, bool]:
    method convert_response (line 291) | def convert_response(self, response: str) -> Hypothesis:
  class KGHypothesis2Experiment (line 307) | class KGHypothesis2Experiment(FactorAndModelHypothesis2Experiment):
    method prepare_context (line 308) | def prepare_context(self, hypothesis: Hypothesis, trace: Trace) -> Tup...
    method convert_feature_experiment (line 348) | def convert_feature_experiment(self, response: str, hypothesis: Hypoth...
    method convert_model_experiment (line 376) | def convert_model_experiment(self, response: str, hypothesis: Hypothes...
    method convert_response (line 412) | def convert_response(self, response: str, hypothesis: Hypothesis, trac...
  class KGTrace (line 419) | class KGTrace(Trace[KGScenario, KGKnowledgeGraph]):

FILE: rdagent/scenarios/kaggle/tpl_ex/aerial-cactus-identification/ensemble.py
  function ensemble_workflow (line 6) | def ensemble_workflow(test_pred_l: list[np.ndarray], val_pred_l: list[np...

FILE: rdagent/scenarios/kaggle/tpl_ex/aerial-cactus-identification/feature.py
  function feat_eng (line 4) | def feat_eng(

FILE: rdagent/scenarios/kaggle/tpl_ex/aerial-cactus-identification/load_data.py
  function load_test_images (line 12) | def load_test_images(folder):
  function load_images_and_labels (line 23) | def load_images_and_labels(csv_file, image_folder):
  function load_data (line 35) | def load_data() -> tuple[np.ndarray, np.ndarray, np.ndarray, list[str]]:

FILE: rdagent/scenarios/kaggle/tpl_ex/aerial-cactus-identification/model01.py
  function model_workflow (line 21) | def model_workflow(

FILE: rdagent/scenarios/qlib/developer/factor_runner.py
  class QlibFactorRunner (line 25) | class QlibFactorRunner(CachedRunner[QlibFactorExperiment]):
    method calculate_information_coefficient (line 35) | def calculate_information_coefficient(
    method deduplicate_new_factors (line 46) | def deduplicate_new_factors(self, SOTA_feature: pd.DataFrame, new_feat...
    method develop (line 64) | def develop(self, exp: QlibFactorExperiment) -> QlibFactorExperiment:

FILE: rdagent/scenarios/qlib/developer/feedback.py
  function process_results (line 24) | def process_results(current_result, sota_result):
  class QlibFactorExperiment2Feedback (line 54) | class QlibFactorExperiment2Feedback(Experiment2Feedback):
    method generate_feedback (line 55) | def generate_feedback(self, exp: Experiment, trace: Trace) -> Hypothes...
  class QlibModelExperiment2Feedback (line 121) | class QlibModelExperiment2Feedback(Experiment2Feedback):
    method generate_feedback (line 122) | def generate_feedback(self, exp: Experiment, trace: Trace) -> Hypothes...

FILE: rdagent/scenarios/qlib/developer/model_runner.py
  class QlibModelRunner (line 14) | class QlibModelRunner(CachedRunner[QlibModelExperiment]):
    method develop (line 28) | def develop(self, exp: QlibModelExperiment) -> QlibModelExperiment:

FILE: rdagent/scenarios/qlib/developer/utils.py
  function _build_base_feature_workspaces (line 14) | def _build_base_feature_workspaces(exp: QlibFactorExperiment) -> list[Fa...
  function _build_execute_calls (line 29) | def _build_execute_calls(exp: QlibFactorExperiment, base_feature_workspa...
  function _resolve_index_level_values (line 44) | def _resolve_index_level_values(df: pd.DataFrame, level_name: str) -> pd...
  function _normalize_factor_index (line 68) | def _normalize_factor_index(df: pd.DataFrame) -> pd.DataFrame | None:
  function _format_index_info (line 94) | def _format_index_info(df: pd.DataFrame | None) -> str:
  function _process_message_and_df (line 100) | def _process_message_and_df(
  function process_factor_data (line 131) | def process_factor_data(exp_or_list: List[QlibFactorExperiment] | QlibFa...

FILE: rdagent/scenarios/qlib/experiment/factor_experiment.py
  class QlibFactorExperiment (line 19) | class QlibFactorExperiment(FactorExperiment[FactorTask, QlibFBWorkspace,...
    method __init__ (line 20) | def __init__(self, *args, **kwargs) -> None:
  class QlibFactorScenario (line 30) | class QlibFactorScenario(Scenario):
    method __init__ (line 31) | def __init__(self) -> None:
    method background (line 56) | def background(self) -> str:
    method get_source_data_desc (line 59) | def get_source_data_desc(self, task: Task | None = None) -> str:
    method output_format (line 63) | def output_format(self) -> str:
    method interface (line 67) | def interface(self) -> str:
    method simulator (line 71) | def simulator(self) -> str:
    method rich_style_description (line 75) | def rich_style_description(self) -> str:
    method experiment_setting (line 79) | def experiment_setting(self) -> str:
    method get_scenario_all_desc (line 82) | def get_scenario_all_desc(
    method get_runtime_environment (line 101) | def get_runtime_environment(self):

FILE: rdagent/scenarios/qlib/experiment/factor_from_report_experiment.py
  class QlibFactorFromReportScenario (line 7) | class QlibFactorFromReportScenario(QlibFactorScenario):
    method __init__ (line 8) | def __init__(self) -> None:
    method rich_style_description (line 13) | def rich_style_description(self) -> str:

FILE: rdagent/scenarios/qlib/experiment/model_experiment.py
  class QlibModelExperiment (line 18) | class QlibModelExperiment(ModelExperiment[ModelTask, QlibFBWorkspace, Mo...
    method __init__ (line 19) | def __init__(self, *args, **kwargs) -> None:
  class QlibModelScenario (line 26) | class QlibModelScenario(Scenario):
    method __init__ (line 27) | def __init__(self) -> None:
    method background (line 50) | def background(self) -> str:
    method source_data (line 54) | def source_data(self) -> str:
    method output_format (line 58) | def output_format(self) -> str:
    method interface (line 62) | def interface(self) -> str:
    method simulator (line 66) | def simulator(self) -> str:
    method rich_style_description (line 70) | def rich_style_description(self) -> str:
    method experiment_setting (line 74) | def experiment_setting(self) -> str:
    method get_scenario_all_desc (line 77) | def get_scenario_all_desc(
    method get_runtime_environment (line 90) | def get_runtime_environment(self):

FILE: rdagent/scenarios/qlib/experiment/quant_experiment.py
  class QlibFactorExperiment (line 29) | class QlibFactorExperiment(FactorExperiment[FactorTask, QlibFBWorkspace,...
    method __init__ (line 30) | def __init__(self, *args, **kwargs) -> None:
  class QlibModelExperiment (line 35) | class QlibModelExperiment(ModelExperiment[ModelTask, QlibFBWorkspace, Mo...
    method __init__ (line 36) | def __init__(self, *args, **kwargs) -> None:
  class QlibQuantScenario (line 41) | class QlibQuantScenario(Scenario):
    method __init__ (line 42) | def __init__(self) -> None:
    method background (line 58) | def background(self, tag=None) -> str:
    method get_source_data_desc (line 82) | def get_source_data_desc(self) -> str:
    method output_format (line 85) | def output_format(self, tag=None) -> str:
    method interface (line 101) | def interface(self, tag=None) -> str:
    method simulator (line 117) | def simulator(self, tag=None) -> str:
    method rich_style_description (line 130) | def rich_style_description(self) -> str:
    method experiment_setting (line 134) | def experiment_setting(self) -> str:
    method get_scenario_all_desc (line 137) | def get_scenario_all_desc(
    method get_runtime_environment (line 187) | def get_runtime_environment(self, tag: str = None) -> str:

FILE: rdagent/scenarios/qlib/experiment/utils.py
  function generate_data_folder_from_qlib (line 13) | def generate_data_folder_from_qlib():
  function get_file_desc (line 54) | def get_file_desc(p: Path, variable_list=[]) -> str:
  function get_data_folder_intro (line 147) | def get_data_folder_intro(fname_reg: str = ".*", flags=0, variable_mappi...

FILE: rdagent/scenarios/qlib/experiment/workspace.py
  class QlibFBWorkspace (line 13) | class QlibFBWorkspace(FBWorkspace):
    method __init__ (line 14) | def __init__(self, template_folder_path: Path, *args, **kwargs) -> None:
    method execute (line 18) | def execute(self, qlib_config_name: str = "conf.yaml", run_env: dict =...

FILE: rdagent/scenarios/qlib/factor_experiment_loader/json_loader.py
  class FactorExperimentLoaderFromDict (line 15) | class FactorExperimentLoaderFromDict(FactorExperimentLoader):
    method load (line 16) | def load(self, factor_dict: dict) -> QlibFactorExperiment:
  class FactorExperimentLoaderFromJsonFile (line 31) | class FactorExperimentLoaderFromJsonFile(FactorExperimentLoader):
    method load (line 32) | def load(self, json_file_path: Path) -> list:
  class FactorExperimentLoaderFromJsonString (line 38) | class FactorExperimentLoaderFromJsonString(FactorExperimentLoader):
    method load (line 39) | def load(self, json_string: str) -> list:
  class FactorTestCaseLoaderFromJsonFile (line 46) | class FactorTestCaseLoaderFromJsonFile:
    method load (line 47) | def load(self, json_file_path: Path) -> TestCases:

FILE: rdagent/scenarios/qlib/factor_experiment_loader/pdf_loader.py
  function classify_report_from_dict (line 29) | def classify_report_from_dict(
  function __extract_factors_name_and_desc_from_content (line 115) | def __extract_factors_name_and_desc_from_content(
  function __extract_factors_formulation_from_content (line 141) | def __extract_factors_formulation_from_content(
  function __extract_factor_and_formulation_from_one_report (line 182) | def __extract_factor_and_formulation_from_one_report(
  function extract_factors_from_report_dict (line 219) | def extract_factors_from_report_dict(
  function merge_file_to_factor_dict_to_factor_dict (line 249) | def merge_file_to_factor_dict_to_factor_dict(
  function __check_factor_dict_relevance (line 270) | def __check_factor_dict_relevance(
  function check_factor_relevance (line 281) | def check_factor_relevance(
  function __check_factor_dict_viability_simulate_json_mode (line 313) | def __check_factor_dict_viability_simulate_json_mode(
  function check_factor_viability (line 324) | def check_factor_viability(
  function __check_factor_duplication_simulate_json_mode (line 356) | def __check_factor_duplication_simulate_json_mode(
  function __kmeans_embeddings (line 397) | def __kmeans_embeddings(embeddings: np.ndarray, k: int = 20) -> list[lis...
  function __deduplicate_factor_dict (line 453) | def __deduplicate_factor_dict(factor_dict: dict[str, dict[str, str]]) ->...
  function deduplicate_factors_by_llm (line 513) | def deduplicate_factors_by_llm(  # noqa: C901, PLR0912
  class FactorExperimentLoaderFromPDFfiles (line 567) | class FactorExperimentLoaderFromPDFfiles(FactorExperimentLoader):
    method load (line 568) | def load(self, file_or_folder_path: str) -> QlibFactorExperiment:

FILE: rdagent/scenarios/qlib/proposal/bandit.py
  class Metrics (line 11) | class Metrics:
    method as_vector (line 21) | def as_vector(self) -> np.ndarray:
  function extract_metrics_from_experiment (line 36) | def extract_metrics_from_experiment(experiment) -> Metrics:
  class LinearThompsonTwoArm (line 55) | class LinearThompsonTwoArm:
    method __init__ (line 56) | def __init__(self, dim: int, prior_var: float = 1.0, noise_var: float ...
    method sample_reward (line 69) | def sample_reward(self, arm: str, x: np.ndarray) -> float:
    method update (line 84) | def update(self, arm: str, x: np.ndarray, r: float) -> None:
    method next_arm (line 90) | def next_arm(self, x: np.ndarray) -> str:
  class EnvController (line 95) | class EnvController:
    method __init__ (line 96) | def __init__(self, weights: Tuple[float, ...] = None) -> None:
    method reward (line 100) | def reward(self, m: Metrics) -> float:
    method decide (line 103) | def decide(self, m: Metrics) -> str:
    method record (line 107) | def record(self, m: Metrics, arm: str) -> None:

FILE: rdagent/scenarios/qlib/proposal/factor_proposal.py
  class QlibFactorHypothesisGen (line 15) | class QlibFactorHypothesisGen(FactorHypothesisGen):
    method __init__ (line 16) | def __init__(self, scen: Scenario) -> Tuple[dict, bool]:
    method prepare_context (line 19) | def prepare_context(self, trace: Trace) -> Tuple[dict, bool]:
    method convert_response (line 48) | def convert_response(self, response: str) -> Hypothesis:
  class QlibFactorHypothesis2Experiment (line 61) | class QlibFactorHypothesis2Experiment(FactorHypothesis2Experiment):
    method prepare_context (line 62) | def prepare_context(self, hypothesis: Hypothesis, trace: Trace) -> Tup...
    method convert_response (line 94) | def convert_response(self, response: str, hypothesis: Hypothesis, trac...

FILE: rdagent/scenarios/qlib/proposal/model_proposal.py
  class QlibModelHypothesisGen (line 14) | class QlibModelHypothesisGen(ModelHypothesisGen):
    method __init__ (line 15) | def __init__(self, scen: Scenario) -> Tuple[dict, bool]:
    method prepare_context (line 18) | def prepare_context(self, trace: Trace) -> Tuple[dict, bool]:
    method convert_response (line 60) | def convert_response(self, response: str) -> Hypothesis:
  class QlibModelHypothesis2Experiment (line 73) | class QlibModelHypothesis2Experiment(ModelHypothesis2Experiment):
    method prepare_context (line 74) | def prepare_context(self, hypothesis: Hypothesis, trace: Trace) -> Tup...
    method convert_response (line 134) | def convert_response(self, response: str, hypothesis: Hypothesis, trac...

FILE: rdagent/scenarios/qlib/proposal/quant_proposal.py
  class QuantTrace (line 16) | class QuantTrace(Trace):
    method __init__ (line 17) | def __init__(self, scen: Scenario) -> None:
  class QlibQuantHypothesis (line 23) | class QlibQuantHypothesis(Hypothesis):
    method __init__ (line 24) | def __init__(
    method __str__ (line 39) | def __str__(self) -> str:
  class QlibQuantHypothesisGen (line 46) | class QlibQuantHypothesisGen(FactorAndModelHypothesisGen):
    method __init__ (line 47) | def __init__(self, scen: Scenario) -> Tuple[dict, bool]:
    method prepare_context (line 50) | def prepare_context(self, trace: Trace) -> Tuple[dict, bool]:
    method convert_response (line 168) | def convert_response(self, response: str) -> Hypothesis:

FILE: rdagent/scenarios/rl/autorl_bench/agents/example_agent/train.py
  function extract_answer (line 16) | def extract_answer(text):
  function load_data (line 34) | def load_data(file_path, ratio=1.0):
  function gsm8k_reward_func (line 47) | def gsm8k_reward_func(completions, answer, **kwargs):
  function submit_for_grading (line 59) | def submit_for_grading(grading_url: str, model_path: str) -> dict | None:
  function main (line 71) | def main():

FILE: rdagent/scenarios/rl/autorl_bench/agents/registry.py
  class Agent (line 14) | class Agent:
    method __post_init__ (line 20) | def __post_init__(self):
  function get_agent (line 24) | def get_agent(agent_id: str) -> Agent:
  function list_agents (line 41) | def list_agents() -> list[str]:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/__init__.py
  class BenchmarkConfig (line 19) | class BenchmarkConfig:
  function get_benchmark (line 114) | def get_benchmark(benchmark_id: str) -> BenchmarkConfig:
  function get_evaluator (line 122) | def get_evaluator(benchmark_id: str) -> BaseEvaluator:
  function list_benchmarks (line 134) | def list_benchmarks() -> list[str]:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/alfworld/data.py
  function _run_alfworld_download (line 14) | def _run_alfworld_download() -> None:
  function _ensure_alfworld_data (line 26) | def _ensure_alfworld_data() -> Path:
  function download_train_data (line 58) | def download_train_data(target_dir: Path) -> None:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/alfworld/eval.py
  class _Tee (line 26) | class _Tee:
    method __init__ (line 29) | def __init__(self, filepath):
    method write (line 33) | def write(self, message):
    method flush (line 38) | def flush(self):
    method isatty (line 42) | def isatty(self):
    method fileno (line 45) | def fileno(self):
  function _log (line 49) | def _log(msg: str):
  function process_ob (line 69) | def process_ob(ob: str) -> str:
  function alfworld_run (line 76) | def alfworld_run(llm_fn: Callable, env, prompt: str, ob: str, max_steps:...
  function create_llm_fn (line 113) | def create_llm_fn(backend: str, model_path: str, **kwargs) -> tuple:
  class ALFWorldEvaluator (line 196) | class ALFWorldEvaluator(BaseEvaluator):
    method __init__ (line 209) | def __init__(self, config):
    method run_eval (line 214) | def run_eval(
    method _ensure_alfworld_data (line 354) | def _ensure_alfworld_data():
    method _expand_env_vars (line 365) | def _expand_env_vars(self, obj):

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/alpacaeval/data.py
  function _convert_row (line 19) | def _convert_row(row: dict) -> dict:
  function download_train_data (line 33) | def download_train_data(target_dir: Path) -> None:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/alpacaeval/eval.py
  class AlpacaEvalEvaluator (line 23) | class AlpacaEvalEvaluator(BaseEvaluator):
    method __init__ (line 26) | def __init__(self, config):
    method run_eval (line 31) | def run_eval(
    method _load_reference_outputs (line 112) | def _load_reference_outputs(self, filename: str) -> List[dict]:
    method _format_prompt (line 123) | def _format_prompt(self, instruction: str, tokenizer) -> str:
    method _generate_model_outputs (line 132) | def _generate_model_outputs(
    method _extract_score (line 181) | def _extract_score(self, df_leaderboard, name: str) -> tuple[float, di...

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/deepsearchqa/data.py
  function load_source_dataset (line 17) | def load_source_dataset() -> Dataset:
  function split_dataset (line 22) | def split_dataset(dataset: Dataset) -> tuple[Dataset, Dataset]:
  function download_train_data (line 32) | def download_train_data(target_dir: Path):

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/deepsearchqa/eval.py
  class DeepSearchQAEvaluator (line 49) | class DeepSearchQAEvaluator(BaseEvaluator):
    method __init__ (line 60) | def __init__(self, config):
    method run_eval (line 65) | def run_eval(self, model_path: str, workspace_path: str, **kwargs) -> ...
    method _react_loop (line 185) | def _react_loop(
    method _get_search_function (line 283) | def _get_search_function(self):
    method _serpapi_search (line 296) | def _serpapi_search(self, query: str, api_key: str) -> str:
    method _duckduckgo_search (line 310) | def _duckduckgo_search(self, query: str) -> str:
    method _judge_answer (line 329) | def _judge_answer(
    method _string_match (line 361) | def _string_match(self, predicted: str, gold: str, answer_type: str) -...

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/gsm8k/data.py
  function download_train_data (line 15) | def download_train_data(target_dir: Path) -> None:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/humaneval/data.py
  function _convert_row (line 19) | def _convert_row(row: dict) -> dict:
  function download_train_data (line 30) | def download_train_data(target_dir: Path) -> None:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/smith/__init__.py
  function discover_smith_benchmarks (line 26) | def discover_smith_benchmarks() -> dict[str, BenchmarkConfig]:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/smith/per_sample_eval.py
  class PerSampleEvaluator (line 19) | class PerSampleEvaluator(BaseEvaluator):
    method __init__ (line 22) | def __init__(self, config):
    method run_eval (line 27) | def run_eval(
  function _cleanup_vllm (line 143) | def _cleanup_vllm(llm) -> None:
  function _apply_range (line 179) | def _apply_range(data: list, test_range: str) -> list:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/webshop/data.py
  function _clone_webshop_repo (line 18) | def _clone_webshop_repo() -> Path:
  function _ensure_repo_in_path (line 36) | def _ensure_repo_in_path():
  function _download_webshop_data (line 61) | def _download_webshop_data():
  function _build_search_index (line 96) | def _build_search_index():
  function download_train_data (line 127) | def download_train_data(target_dir: Path) -> None:

FILE: rdagent/scenarios/rl/autorl_bench/benchmarks/webshop/eval.py
  class _Tee (line 28) | class _Tee:
    method __init__ (line 31) | def __init__(self, filepath):
    method write (line 35) | def write(self, message):
    method flush (line 40) | def flush(self):
    method isatty (line 44) | def isatty(self):
    method fileno (line 47) | def fileno(self):
  function _log (line 51) | def _log(msg: str):
  function create_llm_fn (line 61) | def create_llm_fn(backend: str, model_path: str, **kwargs) -> Tuple[Call...
  function _format_available_actions (line 153) | def _format_available_actions(avail: dict) -> str:
  function build_react_prompt (line 163) | def build_react_prompt(
  function webshop_run (line 201) | def webshop_run(
  class WebShopEvaluator (line 266) | class WebShopEvaluator(BaseEvaluator):
    method __init__ (line 279) | def __init__(self, config):
    method run_eval (line 284) | def run_eval(

FILE: rdagent/scenarios/rl/autorl_bench/conf.py
  class AutoRLBenchSettings (line 14) | class AutoRLBenchSettings(ExtendedBaseSettings):
  function get_autorl_bench_dir (line 30) | def get_autorl_bench_dir() -> Path:
  function get_workspace_dir (line 34) | def get_workspace_dir() -> Path:
  function get_instructions_file (line 38) | def get_instructions_file() -> Path:
  function get_models_dir (line 42) | def get_models_dir() -> Path:
  function get_data_dir (line 46) | def get_data_dir() -> Path:
  function get_baseline_cache_dir (line 50) | def get_baseline_cache_dir() -> Path:

FILE: rdagent/scenarios/rl/autorl_bench/core/evaluator.py
  class EvalResult (line 20) | class EvalResult(TypedDict):
  class BaseEvaluator (line 53) | class BaseEvaluator(ABC):
    method run_eval (line 115) | def run_eval(
    method validate_model (line 140) | def validate_model(self, model_path: str) -> bool:
    method get_default_result (line 144) | def get_default_result(self, benchmark_name: str, model_path: str) -> ...

FILE: rdagent/scenarios/rl/autorl_bench/core/metrics.py
  function _parse_iso_time (line 15) | def _parse_iso_time(value: str) -> Optional[datetime]:
  function _safe_div (line 22) | def _safe_div(numerator: float, denominator: float) -> Optional[float]:
  function compute_metrics (line 28) | def compute_metrics(
  function write_metrics_json (line 110) | def write_metrics_json(workspace: Path, metrics: dict[str, Any]) -> Path:
  function plot_score_trajectory (line 118) | def plot_score_trajectory(workspace: Path, metrics: dict[str, Any]) -> O...
  function run_workspace_metrics (line 145) | def run_workspace_metrics(

FILE: rdagent/scenarios/rl/autorl_bench/core/opencompass.py
  class OpenCompassEvaluator (line 21) | class OpenCompassEvaluator(BaseEvaluator):
    method __init__ (line 28) | def __init__(self, config):
    method run_eval (line 33) | def run_eval(
    method _get_model_inference_config (line 119) | def _get_model_inference_config(self, model_name: str, gpu_count: int)...
    method _parse_results (line 148) | def _parse_results(self, work_dir: Path, result: dict) -> dict:

FILE: rdagent/scenarios/rl/autorl_bench/core/server.py
  function _get_available_gpus (line 25) | def _get_available_gpus() -> Set[str]:
  function _validate_gpu (line 33) | def _validate_gpu(gpu: str, available: Set[str]) -> Optional[str]:
  class GradingServer (line 44) | class GradingServer:
    method __init__ (line 47) | def __init__(
    method _make_cache_key (line 63) | def _make_cache_key(resolved_path: Path) -> str:
    method load_scores (line 77) | def load_scores(self) -> list[dict]:
    method save_scores (line 82) | def save_scores(self, scores: list[dict]):
    method get_evaluator (line 85) | def get_evaluator(self):
    method resolve_model_path (line 91) | def resolve_model_path(self, model_path: str) -> Path:
    method submit (line 113) | def submit(self, model_path: str, gpu: Optional[str] = None) -> dict:
    method set_baseline (line 226) | def set_baseline(self, score: float):
  function get_server (line 236) | def get_server() -> GradingServer:
  function init_server (line 243) | def init_server(task: str, base_model: str, workspace: str) -> GradingSe...
  function submit (line 252) | def submit():
  function health (line 302) | def health():
  function time_status (line 316) | def time_status():
  function set_baseline (line 336) | def set_baseline():
  function run_server (line 349) | def run_server(task: str, base_model: str, workspace: str, host: str = "...
  class GradingServerContext (line 361) | class GradingServerContext:
    method __enter__ (line 364) | def __enter__(self):
    method __exit__ (line 367) | def __exit__(self, *args):
    method get_baseline (line 370) | def get_baseline(self, task: str, model_name: str, model_path: str, wo...
    method load_scores (line 373) | def load_scores(self) -> list:
  class LocalServerContext (line 377) | class LocalServerContext(GradingServerContext):
    method __init__ (line 380) | def __init__(self, task: str, base_model: str, workspace: str, port: i...
    method __enter__ (line 389) | def __enter__(self):
    method __exit__ (line 412) | def __exit__(self, *args):
    method get_baseline (line 417) | def get_baseline(self, task: str, model_name: str, model_path: str, wo...
    method load_scores (line 424) | def load_scores(self) -> list:
  function create_grading_server (line 428) | def create_grading_server(benchmark, workspace: Path, port: int, base_mo...

FILE: rdagent/scenarios/rl/autorl_bench/core/ui.py
  function main (line 16) | def main() -> None:

FILE: rdagent/scenarios/rl/autorl_bench/core/utils.py
  function kill_process_group (line 27) | def kill_process_group(proc: "subprocess.Popen") -> None:
  function ensure_symlink (line 53) | def ensure_symlink(src: Path, dst: Path):
  function download_model (line 68) | def download_model(model_name: str, model_dir: Optional[str] = None) -> ...
  function download_data (line 84) | def download_data(task: str, data_dir: Optional[str] = None) -> str:
  function _safe_model_name (line 143) | def _safe_model_name(model_name: str) -> str:
  function get_baseline_score (line 148) | def get_baseline_score(
  function submit_to_grading_server (line 207) | def submit_to_grading_server(
  function set_baseline_to_server (line 225) | def set_baseline_to_server(score: float, grading_url: Optional[str] = No...
  function init_run_meta (line 241) | def init_run_meta(workspace: Path, timeout_s: int) -> Path:
  function update_run_meta (line 254) | def update_run_meta(workspace: Path, **fields) -> Path:
  function read_run_meta (line 263) | def read_run_meta(workspace: Path) -> dict:
  function setup_workspace (line 269) | def setup_workspace(
  function detect_driver_model (line 331) | def detect_driver_model(env: dict) -> str:
  function append_result (line 336) | def append_result(row: dict) -> Path:
  function print_summary (line 355) | def print_summary(

FILE: rdagent/scenarios/rl/autorl_bench/run.py
  function run (line 40) | def run(
  function main (line 206) | def main():

FILE: rdagent/scenarios/rl/autorl_bench/test/test_benchmark.py
  function main (line 19) | def main():

FILE: rdagent/scenarios/rl/autorl_bench/test/test_fixes.py
  function report (line 26) | def report(name: str, ok: bool, detail: str = ""):
  function test_b1_lora_detection (line 39) | def test_b1_lora_detection():
  function test_b2b3_lock_and_cache (line 168) | def test_b2b3_lock_and_cache():
  function test_b4_error_passthrough (line 254) | def test_b4_error_passthrough():
  function main (line 307) | def main():

FILE: rdagent/scenarios/rl/dev/feedback.py
  class RLExperiment2Feedback (line 11) | class RLExperiment2Feedback(Experiment2Feedback):
    method __init__ (line 14) | def __init__(self, scen: Scenario, version: str = "exp_feedback") -> N...
    method generate_feedback (line 18) | def generate_feedback(
    method _gen_feedback_with_llm (line 51) | def _gen_feedback_with_llm(
    method _gen_error_feedback (line 91) | def _gen_error_feedback(self, hypothesis: str, error_info: str) -> Hyp...

FILE: rdagent/scenarios/rl/experiment/experiment.py
  class RLTask (line 7) | class RLTask(Task):
  class RLExperiment (line 16) | class RLExperiment(Experiment[RLTask, RLWorkspace, RLWorkspace]):
    method __init__ (line 19) | def __init__(self, sub_tasks: list[RLTask], *args, **kwargs) -> None:

FILE: rdagent/scenarios/rl/experiment/workspace.py
  class RLWorkspace (line 19) | class RLWorkspace(FBWorkspace):
    method run (line 22) | def run(self, env: "Env", entry: str) -> EnvResult:

FILE: rdagent/scenarios/rl/loop.py
  class RLPostTrainingRDLoop (line 13) | class RLPostTrainingRDLoop(RDLoop):
    method __init__ (line 20) | def __init__(self, PROP_SETTING: "RLPostTrainingScen"):
    method direct_exp_gen (line 29) | async def direct_exp_gen(self, prev_out: dict[str, Any]):
    method coding (line 36) | def coding(self, prev_out: dict[str, Any]):
    method feedback (line 43) | def feedback(self, prev_out: dict[str, Any]):
    method record (line 54) | def record(self, prev_out: dict[str, Any]):
    method dump (line 60) | def dump(self, path):

FILE: rdagent/scenarios/rl/proposal/proposal.py
  class RLPostTrainingExpGen (line 12) | class RLPostTrainingExpGen(ExpGen):
    method __init__ (line 15) | def __init__(self, scen: Scenario | None = None):
    method gen (line 18) | def gen(self, trace: Trace) -> RLExperiment:
    method _build_trace_summary (line 44) | def _build_trace_summary(self, trace: Trace) -> str:
    method _gen_hypothesis_with_llm (line 64) | def _gen_hypothesis_with_llm(self, trace_summary: str) -> dict:

FILE: rdagent/scenarios/rl/scen/scenario.py
  class RLPostTrainingScen (line 21) | class RLPostTrainingScen(Scenario):
    method __init__ (line 27) | def __init__(self) -> None:
    method background (line 59) | def background(self) -> str:
    method get_runtime_environment (line 77) | def get_runtime_environment(self) -> str:

FILE: rdagent/scenarios/rl/train/runner.py
  class RLPostTrainingRunner (line 23) | class RLPostTrainingRunner(Developer):
    method __init__ (line 26) | def __init__(self, scen: Scenario, timeout: int = 360000) -> None:
    method develop (line 30) | def develop(self, exp: Experiment) -> Experiment:
    method _find_latest_model (line 131) | def _find_latest_model(output_dir: Path) -> Path:

FILE: rdagent/scenarios/shared/get_runtime_info.py
  function get_runtime_environment_by_env (line 9) | def get_runtime_environment_by_env(env: Env) -> str:
  function check_runtime_environment (line 19) | def check_runtime_environment(env: Env) -> str:

FILE: rdagent/scenarios/shared/runtime_info.py
  function get_runtime_info (line 9) | def get_runtime_info():
  function get_gpu_info (line 17) | def get_gpu_info():

FILE: rdagent/utils/__init__.py
  function get_module_by_module_path (line 28) | def get_module_by_module_path(module_path: Union[str, ModuleType]) -> Mo...
  function convert2bool (line 57) | def convert2bool(value: Union[str, bool]) -> bool:
  function try_regex_sub (line 75) | def try_regex_sub(pattern: str, text: str, replace_with: str = "", flags...
  function filter_with_time_limit (line 88) | def filter_with_time_limit(regex_patterns: Union[str, list[str]], text: ...
  function filter_redundant_text (line 100) | def filter_redundant_text(stdout: str) -> str:
  function remove_path_info_from_str (line 193) | def remove_path_info_from_str(base_path: Path, target_string: str) -> str:
  function md5_hash (line 202) | def md5_hash(input_string: str) -> str:

FILE: rdagent/utils/agent/apply_patch.py
  class ActionType (line 21) | class ActionType(str, Enum):
  class FileChange (line 28) | class FileChange:
  class Commit (line 36) | class Commit:
  class DiffError (line 43) | class DiffError(ValueError):
  class Chunk (line 51) | class Chunk:
  class PatchAction (line 58) | class PatchAction:
  class Patch (line 66) | class Patch:
  class Parser (line 74) | class Parser:
    method _cur_line (line 83) | def _cur_line(self) -> str:
    method _norm (line 89) | def _norm(line: str) -> str:
    method is_done (line 94) | def is_done(self, prefixes: tuple[str, ...] | None = None) -> bool:
    method startswith (line 101) | def startswith(self, prefix: str | tuple[str, ...]) -> bool:
    method read_str (line 104) | def read_str(self, prefix: str) -> str:
    method read_line (line 117) | def read_line(self) -> str:
    method parse (line 124) | def parse(self) -> None:
    method _parse_update_file (line 173) | def _parse_update_file(self, text: str) -> PatchAction:
    method _parse_add_file (line 225) | def _parse_add_file(self) -> PatchAction:
  function find_context_core (line 240) | def find_context_core(
  function find_context (line 260) | def find_context(
  function peek_next_section (line 275) | def peek_next_section(
  function _get_updated_file (line 358) | def _get_updated_file(text: str, action: PatchAction, path: str) -> str:
  function patch_to_commit (line 385) | def patch_to_commit(patch: Patch, orig: dict[str, str]) -> Commit:
  function text_to_patch (line 414) | def text_to_patch(text: str, orig: dict[str, str], prefix: Path | None =...
  function identify_files_needed (line 428) | def identify_files_needed(text: str, prefix: Path | None = None) -> list...
  function identify_files_added (line 440) | def identify_files_added(text: str, prefix: Path | None = None) -> list[...
  function load_files (line 453) | def load_files(paths: list[str], open_fn: Callable[[str], str]) -> dict[...
  function apply_commit (line 457) | def apply_commit(
  function process_patch (line 483) | def process_patch(
  function open_file (line 504) | def open_file(path: str) -> str:
  function write_file (line 509) | def write_file(path: str, content: str) -> None:
  function remove_file (line 516) | def remove_file(path: str) -> None:
  function apply_patch_from_text (line 523) | def apply_patch_from_text(patch_text: str, inplace: bool = False, prefix...
  function main (line 535) | def main() -> None:

FILE: rdagent/utils/agent/ret.py
  class AgentOut (line 17) | class AgentOut:
    method get_spec (line 21) | def get_spec(cls, **context: Any) -> str:
    method extract_output (line 25) | def extract_output(cls, resp: str) -> Any:
  class PythonAgentOut (line 29) | class PythonAgentOut(AgentOut):
    method get_spec (line 31) | def get_spec(cls):
    method extract_output (line 35) | def extract_output(cls, resp: str):
  class MarkdownAgentOut (line 45) | class MarkdownAgentOut(AgentOut):
    method get_spec (line 47) | def get_spec(cls):
    method extract_output (line 51) | def extract_output(cls, resp: str):
  class BatchEditOut (line 59) | class BatchEditOut(AgentOut):
    method get_spec (line 63) | def get_spec(cls, with_del=True):
    method extract_output (line 67) | def extract_output(cls, resp: str):
  class PythonBatchEditOut (line 71) | class PythonBatchEditOut(AgentOut):
    method get_spec (line 73) | def get_spec(cls, with_del=True):
    method extract_output (line 77) | def extract_output(cls, resp: str):
  class PythonBatchPatchOut (line 89) | class PythonBatchPatchOut(AgentOut):
    method get_spec (line 91) | def get_spec(cls):
    method extract_output (line 95) | def extract_output(cls, resp: str, prefix: Path | None = None) -> str:

FILE: rdagent/utils/agent/tpl.py
  function get_caller_dir (line 21) | def get_caller_dir(upshift: int = 0) -> Path:
  function load_content (line 33) | def load_content(uri: str, caller_dir: Path | None = None, ftype: str = ...
  class RDAT (line 85) | class RDAT:
    method __init__ (line 91) | def __init__(self, uri: str, ftype: str = "yaml"):
    method r (line 122) | def r(self, **context: Any) -> str:

FILE: rdagent/utils/agent/workflow.py
  function build_cls_from_json_with_retry (line 10) | def build_cls_from_json_with_retry(

FILE: rdagent/utils/env.py
  function extract_dir_name_from_path_config (line 69) | def extract_dir_name_from_path_config(path_str: str) -> str:
  function cleanup_container (line 93) | def cleanup_container(container: docker.models.containers.Container | No...
  function normalize_volumes (line 117) | def normalize_volumes(vols: dict[str, str | dict[str, str]], working_dir...
  function pull_image_with_progress (line 139) | def pull_image_with_progress(image: str) -> None:
  class EnvConf (line 164) | class EnvConf(ExtendedBaseSettings):
    method get_workspace_content_for_hash (line 172) | def get_workspace_content_for_hash(self, local_path: str | Path) -> li...
  class EnvResult (line 212) | class EnvResult:
    method __init__ (line 223) | def __init__(self, stdout: str, exit_code: int, running_time: float):
    method update_stdout (line 229) | def update_stdout(self, stdout: str) -> None:
    method stdout (line 233) | def stdout(self) -> str:
    method hash_full_stdout (line 239) | def hash_full_stdout(self, full_stdout: str) -> str:
    method _get_truncated_stdout (line 243) | def _get_truncated_stdout(self, full_stdout: str) -> str:
  class Env (line 251) | class Env(Generic[ASpecificEnvConf]):
    method __init__ (line 260) | def __init__(self, conf: ASpecificEnvConf):
    method zip_a_folder_into_a_file (line 263) | def zip_a_folder_into_a_file(self, folder_path: str, zip_file_path: st...
    method unzip_a_file_into_a_folder (line 275) | def unzip_a_file_into_a_folder(
    method prepare (line 298) | def prepare(self, *args, **kwargs) -> None:  # type: ignore[no-untyped...
    method check_output (line 303) | def check_output(
    method __run_with_retry (line 322) | def __run_with_retry(
    method run (line 355) | def run(
    method cached_run (line 468) | def cached_run(
    method _run (line 508) | def _run(
    method dump_python_code_run_and_get_results (line 537) | def dump_python_code_run_and_get_results(
    method refresh_env (line 564) | def refresh_env(self) -> None:
  class LocalConf (line 575) | class LocalConf(EnvConf):
  class LocalEnv (line 586) | class LocalEnv(Env[ASpecificLocalConf]):
    method prepare (line 591) | def prepare(self) -> None: ...
    method _run (line 593) | def _run(
  class CondaConf (line 738) | class CondaConf(LocalConf):
    method change_bin_path (line 743) | def change_bin_path(self, **data: Any) -> "CondaConf":
    method _update_bin_path (line 747) | def _update_bin_path(self) -> None:
  class MLECondaConf (line 762) | class MLECondaConf(CondaConf):
  class DockerConf (line 767) | class DockerConf(EnvConf):
    method populate_exclude_chmod_paths (line 811) | def populate_exclude_chmod_paths(self) -> "DockerConf":
  class QlibCondaConf (line 832) | class QlibCondaConf(CondaConf):
  class QlibCondaEnv (line 839) | class QlibCondaEnv(LocalEnv[QlibCondaConf]):
    method prepare (line 840) | def prepare(self) -> None:
  function _sync_conda_cache_with_real_envs (line 877) | def _sync_conda_cache_with_real_envs() -> None:
  function _prepare_conda_env (line 905) | def _prepare_conda_env(env_name: str, requirements_file: Path, python_ve...
  class FTCondaConf (line 931) | class FTCondaConf(CondaConf):
  class FTCondaEnv (line 941) | class FTCondaEnv(LocalEnv[FTCondaConf]):
    method prepare (line 948) | def prepare(self) -> None:
  class BenchmarkCondaConf (line 976) | class BenchmarkCondaConf(CondaConf):
  class BenchmarkCondaEnv (line 987) | class BenchmarkCondaEnv(LocalEnv[BenchmarkCondaConf]):
    method prepare (line 994) | def prepare(self) -> None:
  class QlibDockerConf (line 1009) | class QlibDockerConf(DockerConf):
  class KGDockerConf (line 1032) | class KGDockerConf(DockerConf):
  class DSDockerConf (line 1052) | class DSDockerConf(DockerConf):
  class MLEBDockerConf (line 1071) | class MLEBDockerConf(DockerConf):
  class FTDockerConf (line 1090) | class FTDockerConf(DockerConf):
    method get_workspace_content_for_hash (line 1117) | def get_workspace_content_for_hash(self, local_path: str | Path) -> li...
  class BenchmarkDockerConf (line 1131) | class BenchmarkDockerConf(DockerConf):
  class DockerEnv (line 1159) | class DockerEnv(Env[DockerConf]):
    method prepare (line 1162) | def prepare(self, *args, **kwargs) -> None:  # type: ignore[no-untyped...
    method _gpu_kwargs (line 1240) | def _gpu_kwargs(self, client: docker.DockerClient) -> dict:  # type: i...
    method _generate_log_header (line 1288) | def _generate_log_header(self, entry: str | None = None) -> str:
    method _process_container_logs (line 1308) | def _process_container_logs(self, logs: Iterable[bytes], local_path: s...
    method _run (line 1411) | def _run(
    method refresh_env (line 1493) | def refresh_env(self) -> None:
  class QTDockerEnv (line 1511) | class QTDockerEnv(DockerEnv):
    method __init__ (line 1514) | def __init__(self, conf: DockerConf = QlibDockerConf()):
    method prepare (line 1517) | def prepare(self, *args, **kwargs) -> None:  # type: ignore[no-untyped...
  class KGDockerEnv (line 1531) | class KGDockerEnv(DockerEnv):
    method __init__ (line 1534) | def __init__(self, competition: str | None = None, conf: DockerConf = ...
  class MLEBDockerEnv (line 1538) | class MLEBDockerEnv(DockerEnv):
    method __init__ (line 1541) | def __init__(self, conf: DockerConf = MLEBDockerConf()):
  class FTDockerEnv (line 1545) | class FTDockerEnv(DockerEnv):
    method __init__ (line 1558) | def __init__(self, conf: DockerConf = FTDockerConf()):
  class BenchmarkDockerEnv (line 1562) | class BenchmarkDockerEnv(DockerEnv):
    method __init__ (line 1576) | def __init__(self, conf: DockerConf = BenchmarkDockerConf()):

FILE: rdagent/utils/fmt.py
  function shrink_text (line 6) | def shrink_text(

FILE: rdagent/utils/qlib.py
  function validate_qlib_features (line 199) | def validate_qlib_features(expressions: list[str]) -> bool:

FILE: rdagent/utils/repo/diff.py
  function generate_diff (line 6) | def generate_diff(dir1: str, dir2: str, file_pattern: str = "*.py") -> l...
  function generate_diff_from_dict (line 42) | def generate_diff_from_dict(file_dict1: dict, file_dict2: dict, file_pat...

FILE: rdagent/utils/repo/repo_utils.py
  class RepoAnalyzer (line 8) | class RepoAnalyzer:
    method __init__ (line 9) | def __init__(self, repo_path: str):
    method summarize_repo (line 13) | def summarize_repo(self, verbose_level: int = 1, doc_str_level: int = ...
    method _generate_tree_structure (line 47) | def _generate_tree_structure(self) -> str:
    method _summarize_file (line 65) | def _summarize_file(self, file_path: Path, verbose_level: int, doc_str...
    method _summarize_class (line 89) | def _summarize_class(self, node: ast.ClassDef, verbose_level: int, doc...
    method _summarize_function (line 103) | def _summarize_function(
    method highlight (line 130) | def highlight(self, file_names: Union[str, List[str]]) -> Dict[str, str]:

FILE: rdagent/utils/workflow/loop.py
  class LoopMeta (line 33) | class LoopMeta(type):
    method _get_steps (line 36) | def _get_steps(bases: tuple[type, ...]) -> list[str]:
    method __new__ (line 53) | def __new__(mcs, clsname: str, bases: tuple[type, ...], attrs: dict[st...
  class LoopTrace (line 78) | class LoopTrace:
  class LoopBase (line 85) | class LoopBase:
    class LoopTerminationError (line 110) | class LoopTerminationError(Exception):
    class LoopResumeError (line 113) | class LoopResumeError(Exception):
    method __init__ (line 116) | def __init__(self) -> None:
    method get_unfinished_loop_cnt (line 137) | def get_unfinished_loop_cnt(self, next_loop: int) -> int:
    method get_semaphore (line 144) | def get_semaphore(self, step_name: str) -> asyncio.Semaphore:
    method pbar (line 161) | def pbar(self) -> tqdm:
    method close_pbar (line 167) | def close_pbar(self) -> None:
    method _check_exit_conditions_on_step (line 172) | def _check_exit_conditions_on_step(self, loop_id: Optional[int] = None...
    method _run_step (line 194) | async def _run_step(self, li: int, force_subproc: bool = False) -> None:
    method kickoff_loop (line 313) | async def kickoff_loop(self) -> None:
    method execute_loop (line 336) | async def execute_loop(self) -> None:
    method run (line 353) | async def run(self, step_n: int | None = None, loop_n: int | None = No...
    method withdraw_loop (line 406) | def withdraw_loop(self, loop_idx: int) -> None:
    method dump (line 426) | def dump(self, path: str | Path) -> None:
    method truncate_session_folder (line 434) | def truncate_session_folder(self, li: int, si: int) -> None:
    method load (line 454) | def load(
    method __getstate__ (line 529) | def __getstate__(self) -> dict[str, Any]:
    method __setstate__ (line 539) | def __setstate__(self, state: dict[str, Any]) -> None:
  function kill_subprocesses (line 545) | def kill_subprocesses() -> None:

FILE: rdagent/utils/workflow/misc.py
  function wait_retry (line 8) | def wait_retry(

FILE: rdagent/utils/workflow/tracking.py
  class WorkflowTracker (line 34) | class WorkflowTracker:
    method __init__ (line 42) | def __init__(self, loop_base: "LoopBase"):
    method is_enabled (line 52) | def is_enabled() -> bool:
    method _datetime_to_float (line 57) | def _datetime_to_float(dt: datetime.datetime) -> float:
    method log_workflow_state (line 61) | def log_workflow_state(self) -> None:

FILE: test/finetune/test_benchmark.py
  function run_benchmark_simple (line 28) | def run_benchmark_simple(

FILE: test/finetune/test_benchmark_api.py
  function generate_api_config (line 94) | def generate_api_config(
  function run_benchmark_api (line 180) | def run_benchmark_api(

FILE: test/finetune/test_benchmark_tablebench.py
  function run_benchmark_simple (line 24) | def run_benchmark_simple(

FILE: test/notebook/test_notebook_converter.py
  function normalize_nb_json_for_comparison (line 10) | def normalize_nb_json_for_comparison(nb_json_str):
  class TestNotebookConverter (line 18) | class TestNotebookConverter(unittest.TestCase):
    method __init__ (line 19) | def __init__(self, *args, **kwargs):
    method test_validation_pass (line 24) | def test_validation_pass(self):
    method test_validation_missing_main_fn (line 30) | def test_validation_missing_main_fn(self):
    method test_validation_missing_sections (line 39) | def test_validation_missing_sections(self):
    method test_argparse_happy_path (line 48) | def test_argparse_happy_path(self):
    method test_argparse_with_dupe_sys (line 96) | def test_argparse_with_dupe_sys(self):
    method test_convert (line 149) | def test_convert(self):
    method test_convert_2 (line 166) | def test_convert_2(self):

FILE: test/notebook/test_util.py
  class TestExtractFunctionBody (line 21) | class TestExtractFunctionBody(unittest.TestCase):
    method test_happy_path (line 22) | def test_happy_path(self):
    method test_happy_path_complex (line 42) | def test_happy_path_complex(self):
    method test_empty (line 78) | def test_empty(self):
    method test_missing_func (line 83) | def test_missing_func(self):
  class TestSplitCodeSections (line 98) | class TestSplitCodeSections(unittest.TestCase):
    method test_happy_path (line 99) | def test_happy_path(self):
    method test_happy_path_no_header (line 143) | def test_happy_path_no_header(self):
    method test_wrong_format (line 187) | def test_wrong_format(self):
    method test_empty (line 219) | def test_empty(self):
    method test_single_no_sections (line 226) | def test_single_no_sections(self):
    method test_single_with_section (line 233) | def test_single_with_section(self):
    method test_no_sections (line 240) | def test_no_sections(self):
    method test_ignores_indented_calls (line 268) | def test_ignores_indented_calls(self):
  class TestSplitOutputSections (line 323) | class TestSplitOutputSections(unittest.TestCase):
    method test_happy_path (line 324) | def test_happy_path(self):
    method test_happy_path_no_header (line 356) | def test_happy_path_no_header(self):
    method test_wrong_format (line 388) | def test_wrong_format(self):
    method test_empty (line 413) | def test_empty(self):
    method test_single_no_sections (line 419) | def test_single_no_sections(self):
    method test_single_with_section (line 425) | def test_single_with_section(self):
    method test_no_sections (line 431) | def test_no_sections(self):
    method test_ignore_spaces (line 452) | def test_ignore_spaces(self):
    method test_ignore_unknown_section (line 477) | def test_ignore_unknown_section(self):
  class TestExtractSectionComments (line 521) | class TestExtractSectionComments(unittest.TestCase):
    method test_happy_path (line 522) | def test_happy_path(self):
    method test_happy_path_multiline (line 548) | def test_happy_path_multiline(self):
    method test_no_comment (line 575) | def test_no_comment(self):
    method test_arbitrary_print_happy_path (line 600) | def test_arbitrary_print_happy_path(self):
    method test_empty_string (line 620) | def test_empty_string(self):
  class TestExtractFirstSectionNameFromCode (line 627) | class TestExtractFirstSectionNameFromCode(unittest.TestCase):
    method test_happy_path (line 628) | def test_happy_path(self):
    method test_no_section (line 642) | def test_no_section(self):
    method test_empty_string (line 653) | def test_empty_string(self):
  class TestExtractFirstSectionNameFromOutput (line 659) | class TestExtractFirstSectionNameFromOutput(unittest.TestCase):
    method test_happy_path (line 660) | def test_happy_path(self):
    method test_no_section (line 673) | def test_no_section(self):
    method test_empty_string (line 684) | def test_empty_string(self):
  class TestIsFunctionCalled (line 690) | class TestIsFunctionCalled(unittest.TestCase):
    method test_happy_path (line 691) | def test_happy_path(self):
    method test_happy_path_with_args (line 695) | def test_happy_path_with_args(self):
    method test_happy_path_with_args_multiline (line 703) | def test_happy_path_with_args_multiline(self):
    method test_not_called (line 714) | def test_not_called(self):
    method test_wrong_format (line 724) | def test_wrong_format(self):
    method test_empty_string (line 728) | def test_empty_string(self):
  class TestRemoveFunction (line 733) | class TestRemoveFunction(unittest.TestCase):
    method test_happy_path (line 734) | def test_happy_path(self):
    method test_function_does_not_exist (line 740) | def test_function_does_not_exist(self):
    method test_empty (line 746) | def test_empty(self):
    method test_preserves_comments (line 752) | def test_preserves_comments(self):
  class TestRemoveMainBlock (line 785) | class TestRemoveMainBlock(unittest.TestCase):
    method test_happy_path (line 786) | def test_happy_path(self):
    method test_one_liner (line 797) | def test_one_liner(self):
    method test_happy_path_arbitrary_content (line 807) | def test_happy_path_arbitrary_content(self):
    method test_block_does_not_exist (line 820) | def test_block_does_not_exist(self):
    method test_empty (line 836) | def test_empty(self):
  class TestExtractTopLevelFunctions (line 843) | class TestExtractTopLevelFunctions(unittest.TestCase):
    method test_happy_path (line 844) | def test_happy_path(self):
    method test_empty (line 889) | def test_empty(self):
    method test_stop_at_code (line 894) | def test_stop_at_code(self):
    method test_trailing_comment (line 937) | def test_trailing_comment(self):
  class TestSplitCodeAndOutputIntoSections (line 983) | class TestSplitCodeAndOutputIntoSections(unittest.TestCase):
    method test_happy_path (line 984) | def test_happy_path(self):
    method test_empty_code (line 1087) | def test_empty_code(self):
    method test_empty_outputs (line 1142) | def test_empty_outputs(self):
    method test_ignored_sections (line 1227) | def test_ignored_sections(self):
    method test_complex (line 1347) | def test_complex(self):
  function S (line 1916) | def S(s_arr):

FILE: test/notebook/testfiles/main.py
  function print_eda (line 39) | def print_eda(train_df):
  class CactusDataset (line 61) | class CactusDataset(Dataset):
    method __init__ (line 62) | def __init__(self, image_ids, labels=None, id2path=None, transforms=No...
    method __len__ (line 68) | def __len__(self):
    method __getitem__ (line 71) | def __getitem__(self, idx):
  function get_transforms (line 87) | def get_transforms(mode='train'):
  function get_dataloader (line 137) | def get_dataloader(dataset, batch_size, shuffle=False, num_workers=4, pi...
  function get_efficientnet_b3 (line 146) | def get_efficientnet_b3(dropout_rate=0.3):
  function compute_class_weight (line 155) | def compute_class_weight(y):
  function train_one_epoch (line 171) | def train_one_epoch(model, loss_fn, optimizer, scheduler, dataloader, de...
  function eval_model (line 197) | def eval_model(model, loss_fn, dataloader, device, class_weights):
  function confusion_info (line 223) | def confusion_info(y_true, y_pred, threshold=0.5):
  function inference_and_submission (line 228) | def inference_and_submission(train_df, train_id2path, test_img_ids, test...
  function main (line 300) | def main():

FILE: test/notebook/testfiles/main2.py
  function seed_everything (line 30) | def seed_everything(seed=42):
  function main (line 37) | def main():

FILE: test/notebook/testfiles/main_missing_main_fn.py
  function print_eda (line 39) | def print_eda(train_df):
  class CactusDataset (line 61) | class CactusDataset(Dataset):
    method __init__ (line 62) | def __init__(self, image_ids, labels=None, id2path=None, transforms=No...
    method __len__ (line 68) | def __len__(self):
    method __getitem__ (line 71) | def __getitem__(self, idx):
  function get_transforms (line 87) | def get_transforms(mode='train'):
  function get_dataloader (line 137) | def get_dataloader(dataset, batch_size, shuffle=False, num_workers=4, pi...
  function get_efficientnet_b3 (line 146) | def get_efficientnet_b3(dropout_rate=0.3):
  function compute_class_weight (line 155) | def compute_class_weight(y):
  function train_one_epoch (line 171) | def train_one_epoch(model, loss_fn, optimizer, scheduler, dataloader, de...
  function eval_model (line 197) | def eval_model(model, loss_fn, dataloader, device, class_weights):
  function confusion_info (line 223) | def confusion_info(y_true, y_pred, threshold=0.5):
  function inference_and_submission (line 228) | def inference_and_submission(train_df, train_id2path, test_img_ids, test...

FILE: test/notebook/testfiles/main_missing_sections.py
  function print_eda (line 39) | def print_eda(train_df):
  class CactusDataset (line 61) | class CactusDataset(Dataset):
    method __init__ (line 62) | def __init__(self, image_ids, labels=None, id2path=None, transforms=No...
    method __len__ (line 68) | def __len__(self):
    method __getitem__ (line 71) | def __getitem__(self, idx):
  function get_transforms (line 87) | def get_transforms(mode='train'):
  function get_dataloader (line 137) | def get_dataloader(dataset, batch_size, shuffle=False, num_workers=4, pi...
  function get_efficientnet_b3 (line 146) | def get_efficientnet_b3(dropout_rate=0.3):
  function compute_class_weight (line 155) | def compute_class_weight(y):
  function train_one_epoch (line 171) | def train_one_epoch(model, loss_fn, optimizer, scheduler, dataloader, de...
  function eval_model (line 197) | def eval_model(model, loss_fn, dataloader, device, class_weights):
  function confusion_info (line 223) | def confusion_info(y_true, y_pred, threshold=0.5):
  function inference_and_submission (line 228) | def inference_and_submission(train_df, train_id2path, test_img_ids, test...
  function main (line 300) | def main():

FILE: test/oai/test_advanced.py
  function _worker (line 14) | def _worker(system_prompt, user_prompt):
  class TestAdvanced (line 22) | class TestAdvanced(unittest.TestCase):
    method test_chat_cache_multiprocess (line 24) | def test_chat_cache_multiprocess(self) -> None:
    method test_chat_multi_round (line 72) | def test_chat_multi_round(self) -> None:
    method test_chat_cache (line 89) | def test_chat_cache(self) -> None:

FILE: test/oai/test_base.py
  class MockBackend (line 4) | class MockBackend:
    method __init__ (line 5) | def __init__(self):
    method _add_json_in_prompt (line 8) | def _add_json_in_prompt(self, new_messages):
  function test_json_added_once (line 12) | def test_json_added_once():

FILE: test/oai/test_completion.py
  class TestPersonModel (line 10) | class TestPersonModel(BaseModel):
  class TestChatCompletion (line 18) | class TestChatCompletion(unittest.TestCase):
    method test_chat_completion (line 19) | def test_chat_completion(self) -> None:
    method test_chat_completion_json_mode (line 29) | def test_chat_completion_json_mode(self) -> None:
    method test_build_messages_and_calculate_token (line 41) | def test_build_messages_and_calculate_token(self) -> None:
    method test_json_mode_with_specific_target_type (line 48) | def test_json_mode_with_specific_target_type(self) -> None:
    method test_response_format_with_basemodel (line 68) | def test_response_format_with_basemodel(self) -> None:

FILE: test/oai/test_embedding_and_similarity.py
  class TestEmbedding (line 9) | class TestEmbedding(unittest.TestCase):
    method test_embedding (line 10) | def test_embedding(self) -> None:
    method test_embedding_list (line 16) | def test_embedding_list(self) -> None:
    method test_embedding_similarity (line 22) | def test_embedding_similarity(self) -> None:
    method test_embedding_long_text_truncation (line 29) | def test_embedding_long_text_truncation(self) -> None:

FILE: test/oai/test_llm_connectivity.py
  function test_model (line 33) | def test_model(model: str) -> tuple:

FILE: test/oai/test_prefect_cache.py
  class PydanticTest (line 7) | class PydanticTest(unittest.TestCase):
    method test_context7_cache (line 17) | def test_context7_cache(self):

FILE: test/oai/test_pydantic.py
  class PydanticTest (line 6) | class PydanticTest(unittest.TestCase):
    method test_context7 (line 8) | def test_context7(self):

FILE: test/qlib/test_model_factor_proposal.py
  function mixed_model_trace (line 15) | def mixed_model_trace():
  function mixed_factor_trace (line 29) | def mixed_factor_trace():
  function test_model_proposal_import (line 42) | def test_model_proposal_import():
  function test_factor_proposal_import (line 46) | def test_factor_proposal_import():
  function test_model_filtering (line 50) | def test_model_filtering(mixed_model_trace):
  function test_factor_filtering (line 69) | def test_factor_filtering(mixed_factor_trace):
  function test_code_inspection (line 95) | def test_code_inspection(converter_class, trace_fixture, request, expect...

FILE: test/utils/coder/test_CoSTEER.py
  class CoSTEERTest (line 4) | class CoSTEERTest(unittest.TestCase):
    method setUp (line 6) | def setUp(self):
    method tearDown (line 9) | def tearDown(self):
    method to_str (line 12) | def to_str(self, obj):
    method test_data_loader (line 15) | def test_data_loader(self):
    method test_feature (line 23) | def test_feature(self):
    method test_model (line 30) | def test_model(self):
    method test_ensemble (line 37) | def test_ensemble(self):
    method test_workflow (line 44) | def test_workflow(self):

FILE: test/utils/coder/test_finetune_coder.py
  function develop_one_competition (line 9) | def develop_one_competition():

FILE: test/utils/test_agent_infra.py
  class TestAgentInfra (line 8) | class TestAgentInfra(unittest.TestCase):
    method test_agent_infra (line 9) | def test_agent_infra(self):
    method test_include (line 27) | def test_include(self):

FILE: test/utils/test_conf.py
  class ConfUtils (line 10) | class ConfUtils(unittest.TestCase):
    method test_conf (line 12) | def test_conf(self):
    method test_ds_costeer_conf (line 31) | def test_ds_costeer_conf(self):

FILE: test/utils/test_env.py
  class QlibLocalEnv (line 22) | class QlibLocalEnv(LocalEnv):
    method prepare (line 23) | def prepare(self) -> None:
  class EnvUtils (line 32) | class EnvUtils(unittest.TestCase):
    method setUp (line 33) | def setUp(self):
    method tearDown (line 37) | def tearDown(self):
    method test_local (line 43) | def test_local(self):
    method test_local_simple (line 55) | def test_local_simple(self):
    method test_conda_simple (line 69) | def test_conda_simple(self):
    method test_conda_error (line 78) | def test_conda_error(self):
    method test_docker (line 89) | def test_docker(self):
    method test_run (line 106) | def test_run(self):
    method test_docker_mem (line 129) | def test_docker_mem(self):
    method test_cleanup_container_import (line 146) | def test_cleanup_container_import(self):

FILE: test/utils/test_import.py
  class TestRDAgentImports (line 10) | class TestRDAgentImports(unittest.TestCase):
    method setUpClass (line 12) | def setUpClass(cls):
    method import_all_modules_from_directory (line 17) | def import_all_modules_from_directory(directory):
    method test_import_modules (line 49) | def test_import_modules(self):

FILE: test/utils/test_kaggle.py
  class TestTpl (line 11) | class TestTpl(unittest.TestCase):
    method test_competition_template (line 12) | def test_competition_template(self):

FILE: test/utils/test_misc.py
  class A (line 8) | class A(SingletonBaseClass):
    method __init__ (line 9) | def __init__(self, **kwargs):
    method __str__ (line 13) | def __str__(self) -> str:
    method __repr__ (line 16) | def __repr__(self) -> str:
  class MiscTest (line 21) | class MiscTest(unittest.TestCase):
    method test_singleton (line 22) | def test_singleton(self):

FILE: test/utils/test_ws.py
  class TestFBWorkspace (line 9) | class TestFBWorkspace(unittest.TestCase):
    method setUp (line 14) | def setUp(self) -> None:  # noqa: D401
    method tearDown (line 21) | def tearDown(self) -> None:
    method test_checkpoint_roundtrip (line 27) | def test_checkpoint_roundtrip(self) -> None:

FILE: web/components.d.ts
  type GlobalComponents (line 9) | interface GlobalComponents {

FILE: web/src/constants/qlib.js
  constant ALPHA158 (line 1) | const ALPHA158 = {

FILE: web/src/utils/api.js
  function uploadFile (line 5) | function uploadFile(data, config = {}) {
  function trace (line 21) | function trace(data) {
  function control (line 32) | function control(data) {
  function submitUserInteraction (line 43) | function submitUserInteraction(data) {
  function getStdoutDownloadUrl (line 54) | function getStdoutDownloadUrl(traceId) {

FILE: web/src/utils/crypto.js
  method set (line 9) | set(word, keyStr) {
  method get (line 23) | get(word, keyStr) {

FILE: web/src/utils/snap.svg-min.js
  function c (line 19) | function c(a,b){if(a){if(a.nodeType)return w(a);if(e(a,"array")&&c.set)r...
  function d (line 19) | function d(a,b){if(b){if("#text"==a&&(a=y.doc.createTextNode(b.text||b["...
  function e (line 19) | function e(a,b){return b=A.prototype.toLowerCase.call(b),"finite"==b?isF...
  function f (line 19) | function f(a){if("function"==typeof a||Object(a)!==a)return a;var b=new ...
  function h (line 19) | function h(a,b){for(var c=0,d=a.length;d>c;c++)if(a[c]===b)return a.push...
  function i (line 19) | function i(a,b,c){function d(){var e=Array.prototype.slice.call(argument...
  function j (line 19) | function j(a,b,c,d,e,f){if(null==e){var g=a-c,h=b-d;return g||h?(180+180...
  function k (line 19) | function k(a){return a%360*H/180}
  function l (line 19) | function l(a){return 180*a/H%360}
  function m (line 19) | function m(a){var b=[];return a=a.replace(/(?:^|\s)(\w+)\(([^)]+)\)/g,fu...
  function n (line 19) | function n(a,b){var d=aa(a),e=new c.Matrix;if(d)for(var f=0,g=d.length;g...
  function o (line 19) | function o(a){var b=a.node.ownerSVGElement&&w(a.node.ownerSVGElement)||a...
  function p (line 19) | function p(a){return a.node.ownerSVGElement&&w(a.node.ownerSVGElement)||...
  function q (line 19) | function q(a,b,c){function e(a){if(null==a)return I;if(a==+a)return a;d(...
  function r (line 19) | function r(a){e(a,"array")||(a=Array.prototype.slice.call(arguments,0));...
  function s (line 19) | function s(a){if(a.snap in V)return V[a.snap];var b;try{b=a.ownerSVGElem...
  function t (line 19) | function t(a){this.node=a}
  function u (line 19) | function u(a,b){var c=d(a);b.appendChild(c);var e=w(c);return e}
  function v (line 19) | function v(a,b){var c,e,f,g=v.prototype;if(a&&a.tagName&&"svg"==a.tagNam...
  function w (line 19) | function w(a){return a?a instanceof s||a instanceof t?a:a.tagName&&"svg"...
  function x (line 19) | function x(a,b){for(var c=0,d=a.length;d>c;c++){var e={type:a[c].type,at...
  function d (line 19) | function d(a){var d=a.x-b,e=a.y-c;return d*d+e*e}
  function h (line 19) | function h(a,b){if(null==b){var d=!0;if(b="linearGradient"==a.type||"rad...
  function i (line 19) | function i(a){function b(a,b){var d=o(a.node,b);d=d&&d.match(g),d=d&&d[2...
  function j (line 19) | function j(a){return function(){var b=a?"<"+this.type:"",c=this.node.att...
  function h (line 20) | function h(a,b,c){return function(d){var e=d.slice(a,b);return 1==e.leng...
  function f (line 20) | function f(a,b,c,d,e,f){return null==b&&"[object SVGMatrix]"==g.call(a)?...
  function c (line 20) | function c(a){return a[0]*a[0]+a[1]*a[1]}
  function d (line 20) | function d(a){var b=i.sqrt(c(a));a[0]&&(a[0]/=b),a[1]&&(a[1]/=b)}
  function g (line 20) | function g(d){return function(e){if(b.stop(),e instanceof f&&1==e.node.c...
  function h (line 20) | function h(a){b.stop(),a==+a&&(a+="px"),this.node.style.fontSize=a}
  function i (line 20) | function i(a){for(var b=[],c=a.childNodes,d=0,e=c.length;e>d;d++){var f=...
  function j (line 20) | function j(){return b.stop(),this.node.style.fontSize}
  function b (line 20) | function b(a,b){for(var c=(b-h)/(a-i),d=i;a>d;d++)f[d].offset=+(+h+c*(d-...
  function c (line 20) | function c(c){return function(){b.stop();var d=e.doc.defaultView.getComp...
  function d (line 20) | function d(a){return function(c){b.stop();var d="marker"+a.charAt(0).toU...
  function g (line 20) | function g(a){return a}
  function h (line 20) | function h(a){return function(b){return+b.toFixed(3)+a}}
  function a (line 20) | function a(){this.parentNode.removeChild(this)}
  function d (line 20) | function d(){return this.selectAll("stop")}
  function e (line 20) | function e(a,b){var d=l("stop"),e={offset:+b+"%"};a=c.color(a),e["stop-c...
  function f (line 20) | function f(){if("linearGradient"==this.type){var a=l(this.node,"x1")||0,...
  function g (line 20) | function g(a){var d=a,e=this.stops();if("string"==typeof a&&(d=b("snap.u...
  function i (line 20) | function i(a,c){var d,e=b("snap.util.grad.parse",null,c).firstDefined();...
  function j (line 20) | function j(a,b,h,i,j){var k=c._.make("linearGradient",a);return k.stops=...
  function k (line 20) | function k(a,b,g,h,i,j){var k=c._.make("radialGradient",a);return k.stop...
  function e (line 20) | function e(a){var b=e.ps=e.ps||{};return b[a]?b[a].sleep=100:b[a]={sleep...
  function f (line 20) | function f(a,b,c,d){return null==a&&(a=b=c=d=0),null==b&&(b=a.y,c=a.widt...
  function g (line 20) | function g(){return this.join(",").replace(N,"$1")}
  function h (line 20) | function h(a){var b=L(a);return b.toString=g,b}
  function i (line 20) | function i(a,b,c,d,e,f,g,h,i){return null==i?p(a,b,c,d,e,f,g,h):k(a,b,c,...
  function j (line 20) | function j(c,d){function e(a){return+(+a).toFixed(3)}return a._.cacher(f...
  function k (line 20) | function k(a,b,c,d,e,f,g,h,i){var j=1-i,k=T(j,3),l=T(j,2),m=i*i,n=m*i,o=...
  function l (line 20) | function l(b,c,d,e,g,h,i,j){a.is(b,"array")||(b=[b,c,d,e,g,h,i,j]);var k...
  function m (line 20) | function m(a,b,c){return b>=a.x&&b<=a.x+a.width&&c>=a.y&&c<=a.y+a.height}
  function n (line 20) | function n(a,b){return a=f(a),b=f(b),m(b,a.x,a.y)||m(b,a.x2,a.y)||m(b,a....
  function o (line 20) | function o(a,b,c,d,e){var f=-3*b+9*c-9*d+3*e,g=a*f+6*b-12*c+6*d;return a...
  function p (line 20) | function p(a,b,c,d,e,f,g,h,i){null==i&&(i=1),i=i>1?1:0>i?0:i;for(var j=i...
  function q (line 20) | function q(a,b,c,d,e,f,g,h,i){if(!(0>i||p(a,b,c,d,e,f,g,h)<i)){var j,k=1...
  function r (line 20) | function r(a,b,c,d,e,f,g,h){if(!(S(a,c)<R(e,g)||R(a,c)>S(e,g)||S(b,d)<R(...
  function s (line 20) | function s(a,b,c){var d=l(a),e=l(b);if(!n(d,e))return c?0:[];for(var f=p...
  function t (line 20) | function t(a,b){return v(a,b)}
  function u (line 20) | function u(a,b){return v(a,b,1)}
  function v (line 20) | function v(a,b,c){a=G(a),b=G(b);for(var d,e,f,g,h,i,j,k,l,m,n=c?0:[],o=0...
  function w (line 20) | function w(a,b,c){var d=x(a);return m(d,b,c)&&v(a,[["M",b,c],["H",d.x2+1...
  function x (line 20) | function x(a){var b=e(a);if(b.bbox)return L(b.bbox);if(!a)return f();a=G...
  function y (line 20) | function y(a,b,c,d,e){if(e)return[["M",+a+ +e,b],["l",c-2*e,0],["a",e,e,...
  function z (line 20) | function z(a,b,c,d,e){if(null==e&&null==d&&(d=c),a=+a,b=+b,c=+c,d=+d,nul...
  function A (line 20) | function A(b){var c=e(b),d=String.prototype.toLowerCase;if(c.rel)return ...
  function B (line 20) | function B(b){var c=e(b);if(c.abs)return h(c.abs);if(K(b,"array")&&K(b&&...
  function C (line 20) | function C(a,b,c,d){return[a,b,c,d,c,d]}
  function D (line 20) | function D(a,b,c,d,e,f){var g=1/3,h=2/3;return[g*a+h*c,g*b+h*d,g*e+h*c,g...
  function E (line 20) | function E(b,c,d,e,f,g,h,i,j,k){var l,m=120*Q/180,n=Q/180*(+f||0),o=[],p...
  function F (line 20) | function F(a,b,c,d,e,f,g,h){for(var i,j,k,l,m,n,o,p,q=[],r=[[],[]],s=0;2...
  function G (line 20) | function G(a,b){var c=!b&&e(a);if(!b&&c.curve)return h(c.curve);for(var ...
  function H (line 20) | function H(a,b){if(!b)return a;var c,d,e,f,g,h,i;for(a=G(a),e=0,g=a.leng...
  function I (line 20) | function I(a,b){for(var c=[],d=0,e=a.length;e-2*!b>d;d+=2){var f=[{x:+a[...
  function f (line 21) | function f(a){var b=a[0];switch(b.toLowerCase()){case"t":return[b,0,0];c...
  function g (line 21) | function g(b,c,d){b=b||new a.Matrix,c=c||new a.Matrix,b=a.parseTransform...
  function h (line 21) | function h(a){return a}
  function i (line 21) | function i(a){return function(b){return+b.toFixed(3)+a}}
  function j (line 21) | function j(a){return a.join(" ")}
  function k (line 21) | function k(b){return a.rgb(b[0],b[1],b[2],b[3])}
  function l (line 21) | function l(a){var b,c,d,e,f,g,h=0,i=[];for(b=0,c=a.length;c>b;b++){for(f...
  function m (line 21) | function m(a){for(var b=[],c=0,d=a.length;d>c;c++)for(var e=1,f=a[c].len...
  function n (line 21) | function n(a){return isFinite(a)}
  function o (line 21) | function o(b,c){return a.is(b,"array")&&a.is(c,"array")?b.toString()==c....
  function i (line 21) | function i(i,j,l){(i.originalEvent||i).preventDefault(),k._drag.x=j,k._d...
  function j (line 21) | function j(a,c,d){b("snap.draginit."+k.id,k,a,c,d)}
  function f (line 21) | function f(a){a=a.split(/(?=#)/);var b=new String(a[5]);return b[50]=a[0...