From c600468649325ea326a8a6e99367fd69908481dd Mon Sep 17 00:00:00 2001 From: you-n-g Date: Wed, 5 Mar 2025 22:27:46 +0800 Subject: [PATCH] fix: direct_exp_gen to json_target_type in DSExpGen class (#661) * refactor: Rename direct_exp_gen to json_target_type in DSExpGen class * fix type --- rdagent/scenarios/data_science/proposal/exp_gen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rdagent/scenarios/data_science/proposal/exp_gen.py b/rdagent/scenarios/data_science/proposal/exp_gen.py index 4cae7546..8c3dcdb1 100644 --- a/rdagent/scenarios/data_science/proposal/exp_gen.py +++ b/rdagent/scenarios/data_science/proposal/exp_gen.py @@ -420,7 +420,7 @@ class DSExpGen(ExpGen): user_prompt=user_prompt, system_prompt=system_prompt, json_mode=True, - direct_exp_gen=Dict[str, Dict[str, str]], + json_target_type=dict[str, dict[str, str] | str], # workflow_update may be a string ) ) assert "hypothesis_proposal" in resp_dict, "Hypothesis proposal not provided."