File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -154,6 +154,7 @@ def __init__(
154154 self .evaluator = Evaluator (
155155 self .config .evaluator ,
156156 evaluation_file ,
157+ Path (self .initial_program_path ).suffix ,
157158 self .llm_evaluator_ensemble ,
158159 self .evaluator_prompt_sampler ,
159160 database = self .database ,
Original file line number Diff line number Diff line change @@ -41,12 +41,14 @@ def __init__(
4141 self ,
4242 config : EvaluatorConfig ,
4343 evaluation_file : str ,
44+ suffix : str ,
4445 llm_ensemble : Optional [LLMEnsemble ] = None ,
4546 prompt_sampler : Optional [PromptSampler ] = None ,
4647 database : Optional [ProgramDatabase ] = None ,
4748 ):
4849 self .config = config
4950 self .evaluation_file = evaluation_file
51+ self .program_suffix = suffix
5052 self .llm_ensemble = llm_ensemble
5153 self .prompt_sampler = prompt_sampler
5254 self .database = database
@@ -152,7 +154,7 @@ async def evaluate_program(
152154 last_exception = None
153155 for attempt in range (self .config .max_retries + 1 ):
154156 # Create a temporary file for the program
155- with tempfile .NamedTemporaryFile (suffix = ".py" , delete = False ) as temp_file :
157+ with tempfile .NamedTemporaryFile (suffix = self . program_suffix , delete = False ) as temp_file :
156158 temp_file .write (program_code .encode ("utf-8" ))
157159 temp_file_path = temp_file .name
158160
You can’t perform that action at this time.
0 commit comments