@@ -116,7 +116,7 @@ def __init__(self, report_folder: str = None):
116
116
self .valid_pdf_file_count = 0
117
117
self .current_loop_hypothesis = None
118
118
self .current_loop_exp = None
119
- self .steps = ["propose_hypo_exp" , "propose" , "exp_gen " , "coding" , "running" , "feedback" ]
119
+ self .steps = ["propose_hypo_exp" , "propose" , "direct_exp_gen " , "coding" , "running" , "feedback" ]
120
120
121
121
def propose_hypo_exp (self , prev_out : dict [str , Any ]):
122
122
with logger .tag ("r" ):
@@ -144,12 +144,12 @@ def propose_hypo_exp(self, prev_out: dict[str, Any]):
144
144
def propose (self , prev_out : dict [str , Any ]):
145
145
return self .current_loop_hypothesis
146
146
147
- def exp_gen (self , prev_out : dict [str , Any ]):
148
- return self .current_loop_exp
147
+ def direct_exp_gen (self , prev_out : dict [str , Any ]):
148
+ return { "propose" : self .current_loop_hypothesis , "exp_gen" : self . current_loop_exp }
149
149
150
150
def coding (self , prev_out : dict [str , Any ]):
151
151
with logger .tag ("d" ): # develop
152
- exp = self .coder .develop (prev_out ["exp_gen" ])
152
+ exp = self .coder .develop (prev_out ["direct_exp_gen" ][ " exp_gen" ])
153
153
logger .log_object (exp .sub_workspace_list , tag = "coder result" )
154
154
return exp
155
155
0 commit comments